From e4a4d22537ef45b61f181c5636172501b7e693c5 Mon Sep 17 00:00:00 2001
From: kijai <40791699+kijai@users.noreply.github.com>
Date: Sat, 8 Nov 2025 16:04:55 +0200
Subject: [PATCH] Update nodes_sampler.py

---
 nodes_sampler.py | 15 +++++++++------
 1 file changed, 9 insertions(+), 6 deletions(-)

diff --git a/nodes_sampler.py b/nodes_sampler.py
index 04d6e15..be138aa 100644
--- a/nodes_sampler.py
+++ b/nodes_sampler.py
@@ -1351,12 +1351,15 @@ class WanVideoSampler:
                     z = z * c_in
                     timestep = c_noise
 
-                self.noise_front_pad_num = image_cond_input.shape[1] - z.shape[1]
-                if self.noise_front_pad_num > 0:
-                    pad = torch.zeros((z.shape[0], self.noise_front_pad_num, z.shape[2], z.shape[3]), dtype=z.dtype, device=z.device)
-                    z = torch.concat([pad, z], dim=1)
-                    nonlocal seq_len
-                    seq_len = math.ceil((z.shape[2] * z.shape[3]) / 4 * z.shape[1])
+                if image_cond is not None:
+                    self.noise_front_pad_num = image_cond_input.shape[1] - z.shape[1]
+                    if self.noise_front_pad_num > 0:
+                        pad = torch.zeros((z.shape[0], self.noise_front_pad_num, z.shape[2], z.shape[3]), dtype=z.dtype, device=z.device)
+                        z = torch.concat([pad, z], dim=1)
+                        nonlocal seq_len
+                        seq_len = math.ceil((z.shape[2] * z.shape[3]) / 4 * z.shape[1])
+                else:
+                    self.noise_front_pad_num = 0
 
                 base_params = {
                     'x': [z], # latent