marcelbinz commited on
Commit
192a05e
·
verified ·
1 Parent(s): b0e3a08

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +3 -5
requirements.txt CHANGED
@@ -1,5 +1,3 @@
1
- --extra-index-url https://download.pytorch.org/whl/cu128
2
- torch==2.7.0+cu128
3
- torchvision==0.22.0+cu128
4
- transformers
5
- gradio==4.21.0
 
1
+ gradio==4.21.0
2
+ torch==2.2.2+cu121 # Torrch nightly wheel allowed
3
+ torchvision==0.17.2+cu121