mirror of
https://github.com/ace-step/ACE-Step-1.5.git
synced 2026-07-02 16:37:04 +00:00
28 lines
583 B
Text
28 lines
583 B
Text
# Requirements ROCM
|
|
# Core dependencies
|
|
transformers>=4.51.0,<4.58.0
|
|
diffusers
|
|
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 # Not supported on ROCM - soundfile fallback will be used
|
|
torchao
|
|
toml
|
|
modelscope
|
|
|
|
# Training dependencies (required for LoRA training)
|
|
peft>=0.18.0
|
|
lightning>=2.0.0
|
|
tensorboard>=2.0.0
|
|
xxhash
|
|
|
|
# Local package - install with: pip install -e acestep/third_parts/nano-vllm
|
|
# nano-vllm
|