mirror of
https://github.com/comfyanonymous/ComfyUI.git
synced 2026-01-14 00:00:57 +08:00
Merge branch 'comfyanonymous:master' into master
This commit is contained in:
commit
9d0e804d5d
4
nodes.py
4
nodes.py
@ -1852,6 +1852,10 @@ class ImageBatch:
|
||||
CATEGORY = "image"
|
||||
|
||||
def batch(self, image1, image2):
|
||||
if image1.shape[-1] != image2.shape[-1]:
|
||||
channels = min(image1.shape[-1], image2.shape[-1])
|
||||
image1 = image1[..., :channels]
|
||||
image2 = image2[..., :channels]
|
||||
if image1.shape[1:] != image2.shape[1:]:
|
||||
image2 = comfy.utils.common_upscale(image2.movedim(-1,1), image1.shape[2], image1.shape[1], "bilinear", "center").movedim(1,-1)
|
||||
s = torch.cat((image1, image2), dim=0)
|
||||
|
||||
Loading…
Reference in New Issue
Block a user