ClownRat commited on
Commit
b6750ce
•
1 Parent(s): d6ff507

Update requirements.txt

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
- torch==2.2.0
4
- torchvision==0.17.0
5
  transformers==4.42.3
6
  tokenizers==0.19.1
7
  deepspeed==0.13.1
 
1
  --extra-index-url https://download.pytorch.org/whl/cu118
2
  # basic dependencies
3
+ torch==2.2.0+cu118
4
+ torchvision==0.17.0+cu118
5
  transformers==4.42.3
6
  tokenizers==0.19.1
7
  deepspeed==0.13.1