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)
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(
res.permute(1,2,3,0)[:, :, :, :3] # T, H, W, C
res
)[0]
y = torch.concat([msk, y])
motion_patched = patch_motion(processed_tracks, y, temperature, (4, 16), topk)[None]