fffiloni commited on
Commit
3ea318d
·
1 Parent(s): 89dafb4

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +2 -2
requirements.txt CHANGED
@@ -31,8 +31,8 @@ termcolor==2.2.0
31
  thinc==8.1.10
32
  timm==0.6.12
33
  tokenizers==0.13.2
34
- torch
35
- torchvision
36
  tqdm==4.64.1
37
  transformers==4.26.1
38
  triton
 
31
  thinc==8.1.10
32
  timm==0.6.12
33
  tokenizers==0.13.2
34
+ torch==1.13.1+cu116
35
+ torchvision==0.14.1+cu116
36
  tqdm==4.64.1
37
  transformers==4.26.1
38
  triton