Add ACE-Step XL SFT music experiment

This commit is contained in:
Mikei386 committed 2026-09-08 15:16:15 +02:00
1 parent 2724861224
commit 5e18b7776b
4 files changed
+135 -1

No files matched your search

+37
View File
@@ -0,0 +1,37 @@
services:
music-worker:
image: ghcr.io/ace-step/ace-step-1.5:latest@sha256:95652cd780c78a1b1a7f6f0335530430f0ae53d96c7c12d59f9f39fa23d38567
container_name: mike-ai-music-acestep-test
profiles: ["music-test"]
environment:
ACESTEP_MODE: gradio
ACESTEP_CONFIG_PATH: acestep-v15-xl-sft
ACESTEP_LM_MODEL_PATH: acestep-5Hz-lm-1.7B
ACESTEP_LLM_BACKEND: pt
ACESTEP_INIT_SERVICE: "true"
ACESTEP_INIT_LLM: "true"
ACESTEP_DEVICE: cuda
TOKENIZERS_PARALLELISM: "false"
NVIDIA_VISIBLE_DEVICES: ${ACESTEP_GPU_UUID:?set ACESTEP_GPU_UUID to the RTX 5080 UUID}
deploy:
resources:
reservations:
devices:
- driver: nvidia
device_ids: ["${ACESTEP_GPU_UUID:?set ACESTEP_GPU_UUID to the RTX 5080 UUID}"]
capabilities: [gpu]
ports:
- "127.0.0.1:${ACESTEP_UI_PORT:-7861}:7860"
volumes:
- ${ACESTEP_CHECKPOINTS_DIR:-/data/models/acestep/checkpoints}:/app/checkpoints
- ${ACESTEP_HF_CACHE_DIR:-/data/models/acestep/hf-cache}:/root/.cache/huggingface
- ${ACESTEP_OUTPUT_DIR:-/data/music/acestep}:/app/gradio_outputs
- ${ACESTEP_OUTPUT_DIR:-/data/music/acestep}:/app/output
shm_size: "2gb"
restart: "no"
healthcheck:
test: ["CMD-SHELL", "curl -fsS http://127.0.0.1:7860/ >/dev/null"]
interval: 30s
timeout: 10s
start_period: 300s
retries: 3