mrleo1nid/ACE-Step
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Merge pull request #373 from iackov/fix/extend-mode-shape-mismatch

Browse Source
This commit is contained in:
Gong Junmin
2026-02-15 12:57:54 +08:00
committed by GitHub
parent 4a24cfb662 435e9fd667
commit 1bee4c9f5b
1 changed files with 13 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
acestep/pipeline_ace_step.py
+13
View File
@@ -1046,6 +1046,19 @@ class ACEStepPipeline:
if right_pad_frame_length > 0: if right_pad_frame_length > 0:
padd_list.append(retake_latents[:, :, :, -right_pad_frame_length:]) padd_list.append(retake_latents[:, :, :, -right_pad_frame_length:])
target_latents = torch.cat(padd_list, dim=-1) target_latents = torch.cat(padd_list, dim=-1)
# Fix shape mismatch between target_latents and x0
if target_latents.shape[-1] != x0.shape[-1]:
if target_latents.shape[-1] < x0.shape[-1]:
# Pad with zeros if target_latents is shorter
padding = x0.shape[-1] - target_latents.shape[-1]
target_latents = torch.nn.functional.pad(
target_latents, (0, padding), "constant", 0
)
else:
# Trim if target_latents is longer
target_latents = target_latents[..., :x0.shape[-1]]
assert ( assert (
target_latents.shape[-1] == x0.shape[-1] target_latents.shape[-1] == x0.shape[-1]
), f"{target_latents.shape=} {x0.shape=}" ), f"{target_latents.shape=} {x0.shape=}"