mirror of
https://github.com/comfyanonymous/ComfyUI.git
synced 2026-01-22 12:20:16 +08:00
torch.cat() everything
This commit is contained in:
parent
569c22673f
commit
c5ea042907
4
nodes.py
4
nodes.py
@ -1138,8 +1138,6 @@ class LoadImageBatch:
|
|||||||
RETURN_TYPES = ("IMAGE", "MASK")
|
RETURN_TYPES = ("IMAGE", "MASK")
|
||||||
FUNCTION = "load_images"
|
FUNCTION = "load_images"
|
||||||
|
|
||||||
OUTPUT_IS_LIST = (True, True, )
|
|
||||||
|
|
||||||
def load_images(self, images):
|
def load_images(self, images):
|
||||||
output_images = []
|
output_images = []
|
||||||
output_masks = []
|
output_masks = []
|
||||||
@ -1180,7 +1178,7 @@ class LoadImageBatch:
|
|||||||
output_images.append(image)
|
output_images.append(image)
|
||||||
output_masks.append(mask)
|
output_masks.append(mask)
|
||||||
|
|
||||||
return (output_images, output_masks, )
|
return (torch.cat(output_images), torch.cat(output_masks), )
|
||||||
|
|
||||||
@classmethod
|
@classmethod
|
||||||
def IS_CHANGED(s, images):
|
def IS_CHANGED(s, images):
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user