xinyun/ComfyUI-KJNodes
1
0
Fork 0
mirror of https://git.datalinker.icu/kijai/ComfyUI-KJNodes.git synced 2025-12-10 05:15:05 +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": {
"mask": ("MASK", {"default": None}),
"alpha": ("MASK", {"default": None}),
},
}
@ -760,7 +760,7 @@ Merges channel data into an image.
], dim=-1)
image = image.squeeze(-2)
if alpha is not None:
image = torch.cat([image, alpha], dim=-1)
image = torch.cat([image, alpha.unsqueeze(-1)], dim=-1)
return (image,)
class ImagePadForOutpaintMasked: