xinyun/ComfyUI-KJNodes
1
0
Fork 0
mirror of https://git.datalinker.icu/kijai/ComfyUI-KJNodes.git synced 2025-12-16 00:04:35 +08:00
Code Issues Packages Projects Releases Wiki Activity

Update image_nodes.py

Browse Source
This commit is contained in:
kijai 2024-07-24 13:17:47 +03:00
parent 86dc22a028
commit 44f7e559b9
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
nodes/image_nodes.py
View File

@ -740,7 +740,7 @@ class MergeImageChannels:
}, },
"optional": { "optional": {
"mask": ("MASK", {"default": None}), "alpha": ("MASK", {"default": None}),
}, },
} }
@ -760,7 +760,7 @@ Merges channel data into an image.
], dim=-1) ], dim=-1)
image = image.squeeze(-2) image = image.squeeze(-2)
if alpha is not None: if alpha is not None:
image = torch.cat([image, alpha], dim=-1) image = torch.cat([image, alpha.unsqueeze(-1)], dim=-1)
return (image,) return (image,)
class ImagePadForOutpaintMasked: class ImagePadForOutpaintMasked: