This commit is contained in:
Eugene Fairley 2025-06-28 21:59:04 -07:00
parent c254763d6b
commit 894ccf4190

View File

@ -667,9 +667,10 @@ class WanTrackToVideo:
], ],
dim=1).to(start_image.device) dim=1).to(start_image.device)
print("start vid:", res.permute(1,2,3,0)[:, :, :, :3]) res = res.permute(1,2,3,0)[:, :, :, :3] # T, H, W, C
print("start vid:", res.shape, res)
y = vae.encode( y = vae.encode(
res.permute(1,2,3,0)[:, :, :, :3] # T, H, W, C res
)[0] )[0]
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]