SakuraD commited on
Commit
38a3f72
1 Parent(s): 6fda41a

update packages

Browse files
Files changed (1) hide show
  1. requirements.txt +2 -2
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
+ torch==2.1.1+cu118
10
+ torchvision==0.16.1+cu118
11
  transformers
12
  wheel