From a7794c141e6906731aefb8ff8a011d895d6f4b3a Mon Sep 17 00:00:00 2001 From: Eugene Fairley Date: Sun, 29 Jun 2025 02:02:28 -0700 Subject: [PATCH] 'save' --- comfy/model_base.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/comfy/model_base.py b/comfy/model_base.py index 9a6d84d6e..11624c6b0 100644 --- a/comfy/model_base.py +++ b/comfy/model_base.py @@ -1218,8 +1218,8 @@ class WAN21(BaseModel): print(f"image shape: {image.shape}, mask shape: {mask.shape}") res = torch.cat((mask, image), dim=1) tracks = kwargs.get("tracks", None) - # if tracks is not None: - # res = patch_motion(tracks.to(device), res[0], 220.0, (4, 16), 2)[None] + if tracks is not None: + res = patch_motion(tracks.to(device), res[0], 220.0, (4, 16), 2)[None] return res