Update requirements.txt
Browse files- requirements.txt +6 -4
requirements.txt
CHANGED
|
@@ -1,9 +1,11 @@
|
|
| 1 |
-
|
| 2 |
-
torchaudio==2.2.0+cu121
|
| 3 |
--extra-index-url https://download.pytorch.org/whl/cu121
|
|
|
|
|
|
|
| 4 |
|
| 5 |
transformers==4.44.2
|
| 6 |
gradio==4.44.0
|
| 7 |
huggingface-hub==0.24.6
|
| 8 |
-
|
| 9 |
-
|
|
|
|
|
|
| 1 |
+
# Pin PyTorch for ZeroGPU (fixes build + CUDA)
|
|
|
|
| 2 |
--extra-index-url https://download.pytorch.org/whl/cu121
|
| 3 |
+
torch==2.2.0
|
| 4 |
+
torchaudio==2.2.0
|
| 5 |
|
| 6 |
transformers==4.44.2
|
| 7 |
gradio==4.44.0
|
| 8 |
huggingface-hub==0.24.6
|
| 9 |
+
kenlm==0.2.0
|
| 10 |
+
pyannote-audio==3.1.1
|
| 11 |
+
numpy==1.26.4 # Compatible with Torch 2.2.0 (avoids NumPy crash)
|