298977887/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2025-04-13 11:53:31 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix alpha channel mismatch on destination in ImageCompositeMasked

Browse Source
This commit is contained in:
comfyanonymous 2025-03-31 20:59:12 -04:00
parent 0b4584c741
commit 548457bac4
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
comfy_extras/nodes_mask.py
View File

@ -89,6 +89,9 @@ class ImageCompositeMasked:
def composite(self, destination, source, x, y, resize_source, mask = None):
if destination.shape[-1] < source.shape[-1]:
source = source[...,:destination.shape[-1]]
elif destination.shape[-1] > source.shape[-1]:
destination = torch.nn.functional.pad(destination, (0, 1))
destination[..., -1] = source[..., -1]
destination = destination.clone().movedim(-1, 1)
output = composite(destination, source.movedim(-1, 1), x, y, mask, 1, resize_source).movedim(1, -1)
return (output,)