From 0388ac4309fd11a2001f11b1ba1e0aa088c5909c Mon Sep 17 00:00:00 2001
From: Dustin <6962246+djdarcy@users.noreply.github.com>
Date: Sun, 3 May 2026 23:52:08 -0400
Subject: [PATCH] Defensively truncate/pad nested noise components to match
 latent

When noise.is_nested with a different number of components than
latent_image, truncate extras or pad missing components with
torch.zeros_like, mirroring the denoise_mask handling pattern below.

Addresses CodeRabbit nitpick on #13318.
---
 comfy/samplers.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/comfy/samplers.py b/comfy/samplers.py
index 34d4a9a48..755fb0a71 100755
--- a/comfy/samplers.py
+++ b/comfy/samplers.py
@@ -1008,7 +1008,10 @@ class CFGGuider:
         if latent_image.is_nested:
             li_tensors = latent_image.unbind()
             if noise.is_nested:
-                n_tensors = noise.unbind()
+                # Truncate extra noise components, pad missing ones with zeros
+                n_tensors = list(noise.unbind()[:len(li_tensors)])
+                for i in range(len(n_tensors), len(li_tensors)):
+                    n_tensors.append(torch.zeros_like(li_tensors[i]))
             else:
                 # Noise only covers video -- pad remaining components (audio) with zeros
                 n_tensors = [noise]