From 02f3b4226c459b6efcc8dcc99f730b99eed7a3d2 Mon Sep 17 00:00:00 2001
From: SayanoAI <28544690+SayanoAI@users.noreply.github.com>
Date: Wed, 31 Jul 2024 16:09:28 -0400
Subject: [PATCH] iterate merge image batches

---
 custom_nodes/utils.py | 18 +++++++++++++-----
 1 file changed, 13 insertions(+), 5 deletions(-)

diff --git a/custom_nodes/utils.py b/custom_nodes/utils.py
index 9d4baa1..90e4f5d 100644
--- a/custom_nodes/utils.py
+++ b/custom_nodes/utils.py
@@ -45,7 +45,7 @@ class MergeImageBatches:
         return {
             "required": {
                 "images": ("IMAGE",),
-                "iterate": ("BOOLEAN",{"default": False})
+                "iterate": ("BOOLEAN",{"default": True})
             },
         }
 
@@ -100,10 +100,18 @@ class MergeImageBatches:
             print(f"{iterate=}")
 
             if iterate:
-                if self.__archive__ is None:
-                    with tempfile.NamedTemporaryFile(delete=False) as ntf:
-                        self.__archive__ = ntf.name
-                images = self.iter_images(self.__archive__,images)
+                image = images[0]
+                for new_img in images[1:]:
+                    image = torch.cat([image,new_img])
+                    gc_collect()
+
+                # if self.__archive__ is None:
+                #     with tempfile.NamedTemporaryFile(delete=False) as ntf:
+                #         self.__archive__ = ntf.name
+                # images = self.iter_images(self.__archive__,images)
+                del images
+                images = image
+                gc_collect()
             else:
                 images = torch.cat(images)