This commit is contained in:
Eugene Fairley 2025-06-28 22:15:22 -07:00
parent 894ccf4190
commit af03f5dbe3

View File

@ -672,6 +672,7 @@ class WanTrackToVideo:
y = vae.encode( y = vae.encode(
res res
)[0] )[0]
print("mask shape:", msk.shape, "y shape:", y.shape)
y = torch.concat([msk, y]) y = torch.concat([msk, y])
motion_patched = patch_motion(processed_tracks, y, temperature, (4, 16), topk)[None] motion_patched = patch_motion(processed_tracks, y, temperature, (4, 16), topk)[None]
mask, video = motion_patched[:, 0:4], motion_patched[:, 4:] mask, video = motion_patched[:, 0:4], motion_patched[:, 4:]