mirror of
https://github.com/Comfy-Org/ComfyUI.git
synced 2026-07-23 09:10:51 -05:00
Fix issue blend images with alpha (#13615)
Make ImageBlend and ImageCompositeMasked nodes handle images with different channel counts
This commit is contained in:
@@ -86,6 +86,6 @@ def image_alpha_fix(destination, source):
|
|||||||
if destination.shape[-1] < source.shape[-1]:
|
if destination.shape[-1] < source.shape[-1]:
|
||||||
source = source[...,:destination.shape[-1]]
|
source = source[...,:destination.shape[-1]]
|
||||||
elif destination.shape[-1] > source.shape[-1]:
|
elif destination.shape[-1] > source.shape[-1]:
|
||||||
destination = torch.nn.functional.pad(destination, (0, 1))
|
source = torch.nn.functional.pad(source, (0, 1))
|
||||||
destination[..., -1] = 1.0
|
source[..., -1] = 1.0
|
||||||
return destination, source
|
return destination, source
|
||||||
|
|||||||
Reference in New Issue
Block a user