chenjoya commited on
Commit
748651e
1 Parent(s): 839beab

Update requirements.txt

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