From d905e82a9d027c171d1e09ac7f558b1d0e4e074d Mon Sep 17 00:00:00 2001 From: Eugene Fairley Date: Sat, 28 Jun 2025 23:07:56 -0700 Subject: [PATCH] 'save' --- comfy_extras/nodes_wan.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/comfy_extras/nodes_wan.py b/comfy_extras/nodes_wan.py index 8ffde48c0..d54aaf219 100644 --- a/comfy_extras/nodes_wan.py +++ b/comfy_extras/nodes_wan.py @@ -655,7 +655,7 @@ class WanTrackToVideo: msk = msk.view(1, msk.shape[1] // 4, 4, lat_h, lat_w) # first batch - msk = msk.transpose(1, 2)[0] + msk = msk.transpose(1, 2) zero_frames = torch.zeros(3, 81 - 1, height, width) @@ -672,7 +672,7 @@ class WanTrackToVideo: y = vae.encode( res - )[0] + ) print("mask shape:", msk.shape, "y shape:", y.shape) # y = torch.concat([msk, y])