Spaces:
Running on Zero
Running on Zero
File size: 2,127 Bytes
38eae94 99375d0 38eae94 99375d0 38eae94 99375d0 38eae94 99375d0 38eae94 99375d0 38eae94 99375d0 38eae94 | 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 | """GPU smoke tests — skipped by default. Opt in with: pytest -m gpu
Generates the minimum-viable songs end-to-end through the real ACE-Step
pipeline. Run before each release tag.
Skipped automatically in CI by the pyproject ``addopts = -m 'not gpu'``
default. Requires:
- ``acestep`` package installed (Apple Silicon fork on Mac, upstream on CUDA)
- DiT checkpoint at ``./checkpoints/acestep-v15-xl-sft/`` (~16 GB) — download via
``hf download ACE-Step/acestep-v15-xl-sft --local-dir checkpoints/acestep-v15-xl-sft``
- LM checkpoint at ``./checkpoints/acestep-5Hz-lm-0.6B/`` (~1.4 GB) — download via
``hf download ACE-Step/acestep-5Hz-lm-0.6B --local-dir checkpoints/acestep-5Hz-lm-0.6B``
- A real MPS / CUDA device — CPU inference is functionally untested
"""
from __future__ import annotations
from pathlib import Path
import pytest
pytestmark = pytest.mark.gpu
def test_generate_minimum_song():
"""Smallest end-to-end: 10 s instrumental drone, seed=1, 16 diffusion steps.
Asserts the pipeline produces a non-empty audio file. Wall time on
cold start (handlers + weight loading) should be < 5 min on M5 Max
with checkpoints pre-downloaded; subsequent calls in the same process
are bounded by the diffusion compute itself (~10-30 s for these settings).
"""
from backend import ACEStepStudioBackend
b = ACEStepStudioBackend()
out_path, meta = b.dispatch(
mode="generate",
params={
"prompt": "ambient drone, sine pad, slow swell",
"lyrics": "",
"duration_s": 10,
"instrumental": True,
"seed": 1,
"loras": [],
# Tune for smoke speed: fewer steps, lower CFG, skip LM CoT
"advanced": {"steps": 16, "cfg": 3.0, "audio_format": "wav"},
"lm": {"thinking": False},
"dcw": {},
},
)
p = Path(out_path)
assert p.exists(), f"generated file missing: {out_path}"
assert p.stat().st_size > 0, "generated file is empty"
assert meta["mode"] == "generate"
assert meta["seed"] == 1
assert meta["wall_seconds"] > 0
|