Ace-Step-Munk / requirements.txt
OnyxMunk's picture
fix: pin diffusers to commit with torchao logger fix
02a4fc6
# ACE-Step 1.5 - Hugging Face Space (Linux, GPU)
# Python 3.10 or 3.11. Space provides CUDA; adjust torch line if needed.
safetensors==0.7.0
--extra-index-url https://download.pytorch.org/whl/cu124
torch>=2.5.0
torchvision
torchaudio
transformers>=4.51.0,<4.58.0
# Pin to commit with torchao quantizer logger fix (NameError); 0.36.0 has bug
diffusers @ git+https://github.com/huggingface/diffusers.git@1cdb8723b85f1b427031e390e0bd0bebfe92454e
gradio==6.2.0
matplotlib>=3.7.5
scipy>=1.10.1
soundfile>=0.13.1
loguru>=0.7.3
einops>=0.8.1
accelerate>=1.12.0
fastapi>=0.110.0
uvicorn[standard]>=0.27.0
numba>=0.63.1
vector-quantize-pytorch>=1.27.15
torchcodec>=0.9.1
torchao
toml
huggingface_hub
python-dotenv