Sebastian Krüger
74f69707b8
transformers 4.40+ uses serialized_type_name kwarg in _register_pytree_node() which doesn't exist in PyTorch 2.1. Pin to older version. Also removes the ineffective pytree shim. 🤖 Generated with [Claude Code](https://claude.com/claude-code) Co-Authored-By: Claude <noreply@anthropic.com>
35 lines
685 B
Plaintext
35 lines
685 B
Plaintext
# 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
|