SakuraD commited on
Commit
0686ccc
1 Parent(s): 38a3f72

update packages

Browse files
Files changed (1) hide show
  1. requirements.txt +3 -3
requirements.txt CHANGED
@@ -6,7 +6,7 @@ packaging
6
  pytest
7
  setuptools
8
  timm
9
- torch==2.1.1+cu118
10
- torchvision==0.16.1+cu118
11
  transformers
12
- wheel
 
 
 
6
  pytest
7
  setuptools
8
  timm
 
 
9
  transformers
10
+ wheel
11
+ torch==2.1.1+cu118 --find-links https://download.pytorch.org/whl/cu118
12
+ torchvision==0.16.1+cu118 --find-links https://download.pytorch.org/whl/cu118