From ee5e2949b8ea7604e41ce72f25e59bfbfd850d65 Mon Sep 17 00:00:00 2001
From: Eugene Fairley <ecfairley@ucdavis.edu>
Date: Sat, 28 Jun 2025 23:14:10 -0700
Subject: [PATCH] 'save'

---
 comfy/model_base.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/comfy/model_base.py b/comfy/model_base.py
index 0c71aeed0..26ecd21da 100644
--- a/comfy/model_base.py
+++ b/comfy/model_base.py
@@ -1219,7 +1219,7 @@ class WAN21(BaseModel):
         if tracks is not None:
             res = patch_motion(tracks.to(device), res[0], 220.0, (4, 16), 2)[None]
         
-        return res[None]
+        return res
 
 
     def extra_conds(self, **kwargs):