From ccd89290ed387c3b639ef7f6a9d4295be0a97165 Mon Sep 17 00:00:00 2001
From: MrForExample <62230687+MrForExample@users.noreply.github.com>
Date: Tue, 2 Jan 2024 20:27:46 +0100
Subject: [PATCH] Node FilterZeroMasksAndCorrespondingImages Edge case handle

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

diff --git a/nodes.py b/nodes.py
index d698120..d062c8f 100644
--- a/nodes.py
+++ b/nodes.py
@@ -1864,7 +1864,7 @@ class FilterZeroMasksAndCorrespondingImages:
         
         if also_process_images:
             non_zero_mask_images_out = torch.stack(non_zero_mask_images, dim=0)
-            zero_mask_images_out = torch.stack(zero_mask_images, dim=0)
+            zero_mask_images_out = torch.stack(zero_mask_images, dim=0) if len(zero_mask_images) > 0 else None
         else:
             non_zero_mask_images_out = zero_mask_images_out = None