requirements.txt

# --- Core ML & AI Frameworks ---
# Usamos versões específicas para garantir a reprodutibilidade.
# diffusers está fixado em um commit específico da branch 'main' do GitHub para estabilidade.
torch
torchao
torchvision
torchaudio
transformers
accelerate
safetensors
einops
sentencepiece

git+https://github.com/hkchengrex/MMAudio.git@main

# O formato 'package @ git+...' é o padrão moderno para dependências do Git.
git+https://github.com/huggingface/diffusers.git@main

# --- User Interface ---
#Gradio para uma que contém 'mount_app'
gradio>=5.23.1
gradio[oauth]
fastapi
uvicorn[standard]
pydantic
soundfile


tiktoken
transformers_stream_generator

rotary-embedding-torch

# --- Data Handling & I/O ---
Pillow
numpy
PyYAML

# --- Video & Image Processing ---
opencv-python
imageio
imageio-ffmpeg
av

# --- Hugging Face Hub ---
huggingface_hub

google-generativeai


# Metrics
torchmetrics            # Core module for metric
pycocotools            # COCO-related
torch-fidelity         # FID-related

# Experiment Tracking
moviepy               # WandB Logging Image & Video
imageio                # WandB Logging Image & Video
tabulate               # Logging Table
deepdiff             # Find difference of config

# Testing
parameterized         # Define multiple tests through decorators.

# Notebook
mediapy             # Notebook Visualization

# DevOPs
black              # Code formatting
flake8             # Code style
isort                # Import sorting
pre-commit               # Pre-commit hooks
expecttest           # Pytorch dist tests
hypothesis           # Fix randomness