# Core requirements for Suno GPT environment
# Python 3.10.15 required

# PyTorch and related (install with CUDA 12.4)
torch==2.6.0+cu124
torchvision==0.21.0+cu124
torchaudio==2.6.0+cu124

# ML training frameworks
transformers==4.44.0
wandb==0.21.0
deepspeed==0.17.2
pytorch-lightning==2.5.2

# Flash Attention (install from source)
# flash-attn==2.8.1  # Build from source instead

# Audio processing
nnAudio==0.3.3
auraloss==0.4.0
encodec==0.1.1
descript-audiotools==0.7.1

# NLP and text processing
g2p-en==2.1.0
phonemizer==3.2.1
sentencepiece==0.1.97
tiktoken==0.1.2
better-profanity==0.7.0

# Utilities
einops==0.8.1
torchsde==0.2.6
modal==1.1.3
ninja==1.11.1.4

# Additional dependencies
scipy>=1.7.3
numpy>=1.20.3,<2.0
pandas>=1.3.4
tqdm>=4.62.3
joblib>=1.2.0
ffmpeg-python>=0.2.0

# Suno internal (install from local path)
# suno_utils - install with pip install -e /home/vibert/projects/glockenspiel/suno_utils