From c254763d6ba153e25bdaed0ae71dc52696c374b8 Mon Sep 17 00:00:00 2001 From: Eugene Fairley Date: Sat, 28 Jun 2025 00:13:45 -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 b0b9f1cb9..f545531ea 100644 --- a/comfy_extras/nodes_wan.py +++ b/comfy_extras/nodes_wan.py @@ -659,18 +659,19 @@ class WanTrackToVideo: zero_frames = torch.zeros(3, 81 - 1, height, width) + start_image = start_image.permute(3,0,1,2) * 2 - 1 # C, T, H, W res = torch.concat([ start_image.to(start_image.device), zero_frames ], dim=1).to(start_image.device) - + + print("start vid:", res.permute(1,2,3,0)[:, :, :, :3]) y = vae.encode( res.permute(1,2,3,0)[:, :, :, :3] # T, H, W, C )[0] y = torch.concat([msk, y]) - print('?') motion_patched = patch_motion(processed_tracks, y, temperature, (4, 16), topk)[None] mask, video = motion_patched[:, 0:4], motion_patched[:, 4:] # Add motion features to conditioning