ClownRat commited on
Commit
86a4704
•
1 Parent(s): 77ad7b3

Fix cuda version.

Browse files
Files changed (1) hide show
  1. requirements.txt +2 -2
requirements.txt CHANGED
@@ -1,7 +1,7 @@
1
  --extra-index-url https://download.pytorch.org/whl/cu118
2
  # basic dependencies
3
- torchvision==0.15.2
4
- torch==2.0.1
5
  transformers==4.37.2
6
  tokenizers==0.15.1
7
  deepspeed==0.13.1
 
1
  --extra-index-url https://download.pytorch.org/whl/cu118
2
  # basic dependencies
3
+ torchvision==0.15.2+cu118
4
+ torch==2.0.1+cu118
5
  transformers==4.37.2
6
  tokenizers==0.15.1
7
  deepspeed==0.13.1