From 894ccf41901f0e178e81f98b4280a4721d4e7429 Mon Sep 17 00:00:00 2001 From: Eugene Fairley Date: Sat, 28 Jun 2025 21:59:04 -0700 Subject: [PATCH] save --- comfy_extras/nodes_wan.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/comfy_extras/nodes_wan.py b/comfy_extras/nodes_wan.py index f545531ea..d7118e697 100644 --- a/comfy_extras/nodes_wan.py +++ b/comfy_extras/nodes_wan.py @@ -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]