From fc08d0830cf52ce7e32b1113be330079668cd132 Mon Sep 17 00:00:00 2001
From: Saurabh Mahra <61376305+saurabhmahra91@users.noreply.github.com>
Date: Mon, 28 Apr 2025 14:19:47 +0530
Subject: [PATCH] Option to disable mask from growing during Gaussian blur

Adds an option to disable the mask from growing during Gaussian blur. Gaussian blur smoothens the border as well which leads to growing of the mask.
---
 mask.py | 10 ++++++++--
 1 file changed, 8 insertions(+), 2 deletions(-)

diff --git a/mask.py b/mask.py
index dcc85a2..3998c3a 100644
--- a/mask.py
+++ b/mask.py
@@ -21,6 +21,7 @@ def INPUT_TYPES(s):
             "required": {
                 "mask": ("MASK",),
                 "amount": ("INT", { "default": 6, "min": 0, "max": 256, "step": 1, }),
+                "allow_growth": ("BOOLEAN", { "default": True }),
                 "device": (["auto", "cpu", "gpu"],),
             }
         }
@@ -29,7 +30,7 @@ def INPUT_TYPES(s):
     FUNCTION = "execute"
     CATEGORY = "essentials/mask"
 
-    def execute(self, mask, amount, device):
+    def execute(self, mask, amount, allow_growth, device):
         if amount == 0:
             return (mask,)
 
@@ -38,6 +39,8 @@ def execute(self, mask, amount, device):
         elif "cpu" == device:
             mask = mask.to('cpu')
 
+        mask_bin = (mask > 0)
+
         if amount % 2 == 0:
             amount+= 1
 
@@ -46,10 +49,13 @@ def execute(self, mask, amount, device):
 
         mask = T.functional.gaussian_blur(mask.unsqueeze(1), amount).squeeze(1)
 
+        if not allow_growth:
+            mask = mask * mask_bin
+
         if "gpu" == device or "cpu" == device:
             mask = mask.to(comfy.model_management.intermediate_device())
 
-        return(mask,)
+        return (mask, )
 
 class MaskFlip:
     @classmethod