feng2022 commited on
Commit
f8af932
1 Parent(s): 9e99e2a

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +3 -3
requirements.txt CHANGED
@@ -1,8 +1,8 @@
1
  # Torch
2
  #--find-links https://download.pytorch.org/whl/torch_stable.html
3
- torch==1.11.0
4
- torchvision==0.12.0
5
- torchaudio==0.11.0
6
  setuptools==59.5.0
7
  transformers==4.3.3
8
 
 
1
  # Torch
2
  #--find-links https://download.pytorch.org/whl/torch_stable.html
3
+ torch==1.4.0+cu100
4
+ torchvision==0.11.2+cu100
5
+ torchaudio==0.10.1+cu100
6
  setuptools==59.5.0
7
  transformers==4.3.3
8