File size: 1,352 Bytes
658e790
 
 
2ce587d
 
 
 
 
658e790
 
 
 
 
 
15389e6
658e790
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
57403ec
658e790
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
# Hugging Face Spaces Requirements for MiloMusic
# Based on pyproject.toml dependencies with Spaces-specific adaptations

# Build dependencies for flash-attn compilation (must be installed first)
wheel>=0.37.0
setuptools>=50.0.0
packaging>=21.0

# PyTorch with CUDA 12.4 support (using specific index)
--find-links https://download.pytorch.org/whl/test/cu124
torch==2.5.1
torchvision
torchaudio

# Core YuE dependencies
transformers==4.48.1
omegaconf==2.3.0
einops==0.8.1
numpy>=1.21.0,<2.0.0
sentencepiece==0.2.0
tqdm>=4.60.0
tensorboard==2.19.0
descript-audiotools>=0.7.0
descript-audio-codec==1.0.0
mmgp==3.1.4.post15
scipy==1.15.2

# MiloMusic frontend/API dependencies
gradio==5.25.0
groq==0.22.0
openai==1.76.0
python-dotenv==1.1.0
spaces==0.34.2
xxhash==3.5.0
soundfile==0.13.1
huggingface_hub>=0.28.1

# Audio processing
librosa==0.11.0
pydub==0.25.1
pyloudnorm==0.1.1

# Additional utilities
pydantic==2.11.3
requests==2.32.3
PyYAML==6.0.2
fire==0.7.0

# Google AI dependencies
google==3.0.0
google-api-core==2.24.2
google-api-python-client==2.166.0
google-auth==2.38.0
google-auth-httplib2==0.2.0
google-genai==1.10.0
googleapis-common-protos>=1.60.0

# Performance optimization
triton

# Flash Attention - NOTE: will be installed via packages.txt and pre-build script
# flash-attn (handled separately due to --no-build-isolation requirement)