Spaces:
Running
on
A100
Running
on
A100
Upload requirements.txt
Browse files- requirements.txt +34 -0
requirements.txt
ADDED
@@ -0,0 +1,34 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
# Core dependencies
|
2 |
+
uv
|
3 |
+
hydra-core
|
4 |
+
loguru
|
5 |
+
Pillow
|
6 |
+
pydub
|
7 |
+
torch
|
8 |
+
torchvision
|
9 |
+
|
10 |
+
|
11 |
+
# Transformers and training utilities
|
12 |
+
transformers==4.46.0
|
13 |
+
pytorchvideo==0.1.5
|
14 |
+
deepspeed==0.15.4
|
15 |
+
accelerate==0.34.2
|
16 |
+
numpy==1.26.4
|
17 |
+
opencv-python-headless==4.8.0.76
|
18 |
+
matplotlib
|
19 |
+
|
20 |
+
# Audio
|
21 |
+
soundfile
|
22 |
+
librosa
|
23 |
+
openai-whisper
|
24 |
+
ftfy
|
25 |
+
ffmpeg
|
26 |
+
jiwer
|
27 |
+
einops
|
28 |
+
wandb
|
29 |
+
kaldiio
|
30 |
+
peft==0.14.0
|
31 |
+
|
32 |
+
# Compatibility fix
|
33 |
+
protobuf==3.20.*
|
34 |
+
triton==3.1.0
|