From f394cf588d394c339b309e8d2d5d2bec8b77f201 Mon Sep 17 00:00:00 2001
From: Chris <chris.goringe@gmail.com>
Date: Tue, 17 Oct 2023 08:31:39 +1100
Subject: [PATCH] Allow None as input to LatentBatch (#11)

---
 nodes.py | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/nodes.py b/nodes.py
index 15ba175..19b0cd1 100644
--- a/nodes.py
+++ b/nodes.py
@@ -193,6 +193,11 @@ class LatentBatch:
     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:]: