From 2e69927399cb5765ab32afa8aec0858b5646d1ff Mon Sep 17 00:00:00 2001 From: Saquib Alam Date: Tue, 8 Aug 2023 21:04:47 +0530 Subject: [PATCH] debug --- nodes.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/nodes.py b/nodes.py index 789c147ed..c3ca5845a 100644 --- a/nodes.py +++ b/nodes.py @@ -1449,12 +1449,13 @@ class ImageAlphaComposite: FUNCTION = "alpha_composite" def alpha_composite(self, image1, mask, image2): mask1 = torch.ones_like(image1) + mask = torch.ones_like(mask) - mask mask1[0, :, :, 0] = mask mask1[0, :, :, 1] = mask mask1[0, :, :, 2] = mask print(image1.size(), mask1.size()) - image = image1 * mask1 + image = image1 * mask1 + image2 * (torch.ones_like(mask) - mask) return (image,) class ImageScaleBy: