Pad ace step 1.5 ref audio if not long enough. (#12341)

2026-02-06 21:02:11 -08:00 · 2026-02-06 21:02:11 -08:00 · 17e7df43d1
parent 039955c527
commit 17e7df43d1
1 changed files with 4 additions and 0 deletions
--- a/comfy/model_base.py
+++ b/comfy/model_base.py
@ -1577,6 +1577,10 @@ class ACEStep15(BaseModel):
            else:
                out['is_covers'] = comfy.conds.CONDConstant(False)

+        if refer_audio.shape[2] < noise.shape[2]:
+            pad = comfy.ldm.ace.ace_step15.get_silence_latent(noise.shape[2], device)
+            refer_audio = torch.cat([refer_audio.to(pad), pad[:, :, refer_audio.shape[2]:]], dim=2)
+
        out['refer_audio'] = comfy.conds.CONDRegular(refer_audio)
        return out