Hugging Face's logo

Spaces:
RoFurukawa
/
testingSMOL
Sleeping

App Files Community
RoFurukawa commited on
Commit
596bf71
·
verified ·
1 Parent(s): cc6f8e5

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +6 -3
requirements.txt CHANGED
@@ -1,7 +1,10 @@
1
- # GPU wheel for PyTorch (CUDA 12.1)
2
- torch --index-url https://download.pytorch.org/whl/cu121
 
 
 
3
  transformers>=4.52.0
4
  accelerate
5
  num2words
6
  decord
7
- gradio
 
1
+ # PyTorch + TorchVision (CUDA 12.1 wheels)
2
+ torch==2.4.* --index-url https://download.pytorch.org/whl/cu121
3
+ torchvision==0.19.* --index-url https://download.pytorch.org/whl/cu121
4
+
5
+ # Hugging Face + video backends + UI
6
  transformers>=4.52.0
7
  accelerate
8
  num2words
9
  decord
10
+ gradio