| 9 | app = FastAPI(title="ACEStep Pipeline API") |
| 10 | |
| 11 | class ACEStepInput(BaseModel): |
| 12 | checkpoint_path: str |
| 13 | bf16: bool = True |
| 14 | torch_compile: bool = False |
| 15 | device_id: int = 0 |
| 16 | output_path: Optional[str] = None |
| 17 | audio_duration: float |
| 18 | prompt: str |
| 19 | lyrics: str |
| 20 | infer_step: int |
| 21 | guidance_scale: float |
| 22 | scheduler_type: str |
| 23 | cfg_type: str |
| 24 | omega_scale: float |
| 25 | actual_seeds: List[int] |
| 26 | guidance_interval: float |
| 27 | guidance_interval_decay: float |
| 28 | min_guidance_scale: float |
| 29 | use_erg_tag: bool |
| 30 | use_erg_lyric: bool |
| 31 | use_erg_diffusion: bool |
| 32 | oss_steps: List[int] |
| 33 | guidance_scale_text: float = 0.0 |
| 34 | guidance_scale_lyric: float = 0.0 |
| 35 | |
| 36 | class ACEStepOutput(BaseModel): |
| 37 | status: str |
nothing calls this directly
no outgoing calls
no test coverage detected