audio-flamingo-3 / requirements.txt
SreyanG-NVIDIA's picture
Update requirements.txt
de8e389 verified
raw
history blame contribute delete
393 Bytes
# Core dependencies
uv
hydra-core
loguru
Pillow
pydub
torch
torchvision
# Transformers and training utilities
transformers==4.46.0
pytorchvideo==0.1.5
deepspeed==0.15.4
accelerate==0.34.2
numpy==1.26.4
opencv-python-headless==4.8.0.76
matplotlib
# Audio
soundfile
librosa
openai-whisper
ftfy
ffmpeg
jiwer
einops
wandb
kaldiio
peft==0.14.0
# Compatibility fix
protobuf==3.20.*
triton==3.1.0