From ea3063527bb73edac33a93a0863dfcd1b3eaf646 Mon Sep 17 00:00:00 2001 From: Saquib Alam Date: Tue, 8 Aug 2023 22:50:42 +0530 Subject: [PATCH] bug fix --- nodes.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/nodes.py b/nodes.py index c9f6af566..8f7c1e1f0 100644 --- a/nodes.py +++ b/nodes.py @@ -1351,7 +1351,7 @@ class LoadImage: mask1[0, :, :, 0] = mask mask1[0, :, :, 1] = mask mask1[0, :, :, 2] = mask - smooth_image = smooth_image * mask + smooth_image = smooth_image * mask1 mask = 1.0 - mask else: mask = torch.zeros((64,64), dtype=torch.float32, device="cpu")