Spaces:
Running
on
Zero
Running
on
Zero
Update requirements.txt
Browse files- requirements.txt +3 -2
requirements.txt
CHANGED
|
@@ -4,7 +4,7 @@
|
|
| 4 |
# Core ML/AI Libraries
|
| 5 |
torch>=2.0.0
|
| 6 |
torchaudio>=2.0.0
|
| 7 |
-
transformers>=4.35.0
|
| 8 |
numpy>=1.21.0
|
| 9 |
|
| 10 |
# Audio Processing
|
|
@@ -44,5 +44,6 @@ cffi>=1.15.0
|
|
| 44 |
# tensorboard>=2.13.0
|
| 45 |
# wandb>=0.15.0
|
| 46 |
|
| 47 |
-
# Note:
|
|
|
|
| 48 |
# For different CUDA versions, adjust jax[cuda] accordingly
|
|
|
|
| 4 |
# Core ML/AI Libraries
|
| 5 |
torch>=2.0.0
|
| 6 |
torchaudio>=2.0.0
|
| 7 |
+
transformers>=4.27.4,<4.35.0
|
| 8 |
numpy>=1.21.0
|
| 9 |
|
| 10 |
# Audio Processing
|
|
|
|
| 44 |
# tensorboard>=2.13.0
|
| 45 |
# wandb>=0.15.0
|
| 46 |
|
| 47 |
+
# Note: transformers version locked to <4.35.0 for whisper-jax compatibility
|
| 48 |
+
# For CPU-only systems, replace jax[cuda] with jax[cpu]
|
| 49 |
# For different CUDA versions, adjust jax[cuda] accordingly
|