wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-01-12 07:10:52 +08:00
Code Issues Actions 10 Packages Projects Releases Wiki Activity

torch.cat() everything

Browse Source
This commit is contained in:
space-nuko 2023-05-29 17:04:01 -05:00
parent 569c22673f
commit c5ea042907
1 changed files with 1 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
nodes.py
View File

@ -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):