From 8b306ca6c9ac440a4b455c048c5a656b09aaf2e2 Mon Sep 17 00:00:00 2001
From: Eugene Fairley <ecfairley@ucdavis.edu>
Date: Sun, 29 Jun 2025 01:34:09 -0700
Subject: [PATCH]  save

---
 comfy/model_base.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/comfy/model_base.py b/comfy/model_base.py
index 11624c6b0..f155f8096 100644
--- a/comfy/model_base.py
+++ b/comfy/model_base.py
@@ -1218,10 +1218,10 @@ 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
+        return res[None]
 
 
     def extra_conds(self, **kwargs):