SreyanG-NVIDIA commited on
Commit
a0739ff
·
verified ·
1 Parent(s): af0d2a6

Upload requirements.txt

Browse files
Files changed (1) hide show
  1. 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