audiocraft-ui/requirements.txt

# Core ML - Pin to AudioCraft-compatible versions
torch>=2.1.0,<2.4.0
torchaudio>=2.1.0,<2.4.0
audiocraft>=1.3.0
xformers>=0.0.22

# Pin transformers to version compatible with PyTorch 2.1
# Newer versions use serialized_type_name kwarg not in PyTorch 2.1
transformers>=4.30.0,<4.40.0

# UI
gradio>=4.0.0

# API
fastapi>=0.104.0
uvicorn[standard]>=0.24.0
python-multipart>=0.0.6

# GPU Monitoring
nvidia-ml-py>=12.535.0  # Replaces deprecated pynvml

# Storage
aiosqlite>=0.19.0

# Configuration
pydantic>=2.5.0
pydantic-settings>=2.1.0
pyyaml>=6.0

# Audio Processing
numpy>=1.24.0,<2.0.0  # NumPy 2.x incompatible with current PyTorch
scipy>=1.11.0
librosa>=0.10.0
soundfile>=0.12.0