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

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 -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
 
1
  transformers==4.41.0
2
  peft==0.11.1
3
+ torch==2.3.1+cuda121
4
+ torchvision==0.18.1+cuda121
5
+ torchaudio==2.3.1+cuda121