chenjoya commited on
Commit
9373a8f
1 Parent(s): 40976c4

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +3 -3
requirements.txt CHANGED
@@ -1,5 +1,5 @@
1
  transformers==4.41.0
2
  peft==0.11.1
3
- torch==2.3.1
4
- torchvision==0.18.1
5
- torchaudio==2.3.1
 
1
  transformers==4.41.0
2
  peft==0.11.1
3
+ torch==2.3.1 -i https://download.pytorch.org/whl/cu121
4
+ torchvision==0.18.1 -i https://download.pytorch.org/whl/cu121
5
+ torchaudio==2.3.1 -i https://download.pytorch.org/whl/cu121