tmberooney commited on
Commit
0d104f9
·
1 Parent(s): 8de32f7

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +1 -1
requirements.txt CHANGED
@@ -1,5 +1,5 @@
1
  gradio==4.12.0
2
  huggingface_hub==0.16.4
3
  peft==0.7.1
4
- torch==1.9.0
5
  transformers==4.31.0
 
1
  gradio==4.12.0
2
  huggingface_hub==0.16.4
3
  peft==0.7.1
4
+ torch @ https://download.pytorch.org/whl/cu121/torch-2.1.0%2Bcu121-cp310-cp310-linux_x86_64.whl#sha256=0d4e8c52a1fcf5ed6cfc256d9a370fcf4360958fc79d0b08a51d55e70914df46
5
  transformers==4.31.0