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")