audio-flamingo-3 / requirements.txt
PierrunoYT's picture
Add pydantic and fastapi to requirements for importlib.metadata compatibility
0ff89a1
# Core dependencies
uv
hydra-core
loguru
Pillow
pydub
gradio
torch>=2.0.0
torchvision>=0.15.0
huggingface_hub>=0.16.0
# Fix for importlib.metadata entry_points parsing error
pydantic
fastapi
# Transformers and training utilities
transformers==4.46.0
pytorchvideo==0.1.5
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
# Compatibility fix
protobuf==3.20.*
triton==3.1.0
bitsandbytes