Merge pull request #12 from chrisgoringe/main

Merge branch 'main' of https://github.com/ssitu/ComfyUI_fabric
ssitu · Oct 17, 2023 · 3782eca · 3782eca
2 parents 37c6a04 + edaa8e9
commit 3782eca
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/nodes.py b/nodes.py
@@ -193,6 +193,11 @@ def INPUT_TYPES(s):
     CATEGORY = "FABRIC"
 
     def batch(self, latent1, latent2):
+        if (latent1 is None):
+            return (latent2,)
+        if (latent2 is None):
+            return (latent1,)
+
         lat1 = latent1["samples"]
         lat2 = latent2["samples"]
         if lat1.shape[1:] != lat2.shape[1:]: