From a64d8bb0df21efdd933692f11231dbd4abc8d2dc Mon Sep 17 00:00:00 2001
From: kijai <40791699+kijai@users.noreply.github.com>
Date: Fri, 8 Aug 2025 17:03:01 +0300
Subject: [PATCH] Fix fun control 2.2 timestep scheduling

---
 nodes.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/nodes.py b/nodes.py
index c9277e4..d749afd 100644
--- a/nodes.py
+++ b/nodes.py
@@ -2150,7 +2150,7 @@ class WanVideoSampler:
                             (control_end_percent > 0 and idx == 0 and current_step_percentage >= control_start_percent):
                             image_cond_input = torch.cat([control_latents.to(z), image_cond.to(z)])
                         else:
-                            image_cond_input = torch.cat([torch.zeros_like(image_cond, dtype=dtype), image_cond.to(z)])
+                            image_cond_input = torch.cat([torch.zeros_like(control_latents, dtype=dtype), image_cond.to(z)])
                         if fun_ref_image is not None:
                             fun_ref_input = fun_ref_image.to(z)
                         else: