Imaginethat commited on
Commit
7f8ec99
·
verified ·
1 Parent(s): 53d32a4

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +7 -7
requirements.txt CHANGED
@@ -1,15 +1,15 @@
1
  gradio
2
 
3
- # ---- PyTorch (GPU build) ----
4
- --extra-index-url https://download.pytorch.org/whl/cu121
5
- torch==2.6.0
6
- torchvision==0.21.0
7
- torchaudio==2.6.0
8
 
9
- # ---- AVoCaDO / Qwen Omni inference stack ----
10
  transformers>=4.52.0
11
  accelerate
12
  qwen-omni-utils[decord]
13
  decord
14
  sentencepiece
15
- safetensors
 
1
  gradio
2
 
3
+ # PyTorch (must be >= 2.6 because transformers blocks unsafe torch.load)
4
+ --extra-index-url https://download.pytorch.org/whl/cu124
5
+ torch>=2.6,<2.7
6
+ torchvision>=0.21,<0.22
7
+ torchaudio>=2.6,<2.7
8
 
9
+ # Your stack
10
  transformers>=4.52.0
11
  accelerate
12
  qwen-omni-utils[decord]
13
  decord
14
  sentencepiece
15
+ safetensors