From c5ea042907fbc430beff95b0566765d45069ae3e Mon Sep 17 00:00:00 2001 From: space-nuko <24979496+space-nuko@users.noreply.github.com> Date: Mon, 29 May 2023 17:04:01 -0500 Subject: [PATCH] `torch.cat()` everything --- nodes.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/nodes.py b/nodes.py index 73def68d2..953e63555 100644 --- a/nodes.py +++ b/nodes.py @@ -1138,8 +1138,6 @@ class LoadImageBatch: RETURN_TYPES = ("IMAGE", "MASK") FUNCTION = "load_images" - OUTPUT_IS_LIST = (True, True, ) - def load_images(self, images): output_images = [] output_masks = [] @@ -1180,7 +1178,7 @@ class LoadImageBatch: output_images.append(image) output_masks.append(mask) - return (output_images, output_masks, ) + return (torch.cat(output_images), torch.cat(output_masks), ) @classmethod def IS_CHANGED(s, images):