Skip to content

Commit 17e7df4

Browse files
Pad ace step 1.5 ref audio if not long enough. (#12341)
1 parent 039955c commit 17e7df4

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

comfy/model_base.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1577,6 +1577,10 @@ def extra_conds(self, **kwargs):
15771577
else:
15781578
out['is_covers'] = comfy.conds.CONDConstant(False)
15791579

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

0 commit comments

Comments
 (0)