Upto12forenglish commited on
Commit
401c67e
1 Parent(s): 938dff8

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +3 -4
requirements.txt CHANGED
@@ -1,7 +1,6 @@
1
- --find-links https://download.pytorch.org/whl/torch_stable.html
2
- torch==1.2.0+cu92
3
- --find-links https://download.pytorch.org/whl/torch_stable.html
4
- torchvision==0.4.0+cu92
5
  transformers
6
  accelerate
7
  SentencePiece
 
1
+ pytorch
2
+ torchvision
3
+ cudatoolkit=10.1
 
4
  transformers
5
  accelerate
6
  SentencePiece