Hugging Face's logo

Spaces:
marcelbinz
/
Centaur
Runtime error

App Files Community
1
marcelbinz commited on
Commit
ffa1b3f
·
verified ·
1 Parent(s): 192a05e

Update requirements.txt

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