| 34 | guidance_scale_lyric: float = 0.0 |
| 35 | |
| 36 | class ACEStepOutput(BaseModel): |
| 37 | status: str |
| 38 | output_path: Optional[str] |
| 39 | message: str |
| 40 | |
| 41 | def initialize_pipeline(checkpoint_path: str, bf16: bool, torch_compile: bool, device_id: int) -> ACEStepPipeline: |
| 42 | os.environ["CUDA_VISIBLE_DEVICES"] = str(device_id) |