Fix numerical issues with inpaint masks in certain cases

* masks can have values slightly below 1 after certain operations like scaling, this could break noise masks when they are thresholded/rounded at 1.0

Author: Acly

ai_diffusion/comfy_workflow.py

@@ -1114,6 +1114,9 @@ def expand_mask(self, mask: Output, grow: int, blur: int, kernel="gaussian"):
     def shrink_mask(self, mask: Output, shrink: int, blur: int):
         return self.add("INPAINT_ShrinkMask", 1, mask=mask, shrink=shrink, blur=blur)
 
+    def stabilize_mask(self, mask: Output, epsilon=0.01):
+        return self.add("INPAINT_StabilizeMask", 1, mask=mask, epsilon=epsilon)
+
     def apply_mask(self, image: Output, mask: Output):
         return self.add("ETN_ApplyMaskToImage", 1, image=image, mask=mask)
 

ai_diffusion/resources.py

@@ -48,8 +48,8 @@ class CustomNode(NamedTuple):
         "Inpaint Nodes",
         "comfyui-inpaint-nodes",
         "https://github.com/Acly/comfyui-inpaint-nodes",
-        "4ebbdc840db881c50a38965aa7955b76cec2a4c5",
-        ["INPAINT_LoadFooocusInpaint", "INPAINT_ShrinkMask", "INPAINT_ExpandMask"],
+        "ab62c98d408186324eeb7baff32a31dd7961ea64",
+        ["INPAINT_LoadFooocusInpaint", "INPAINT_ShrinkMask", "INPAINT_StabilizeMask"],
     ),
 ]
 

ai_diffusion/workflow.py

@@ -759,7 +759,9 @@ def scale_to_initial(
     extent: ScaledExtent, w: ComfyWorkflow, image: Output, models: ModelDict, is_mask=False
 ):
     if is_mask and extent.target != extent.initial:
-        return w.scale_mask(image, extent.initial)
+        result = w.scale_mask(image, extent.initial)
+        result = w.stabilize_mask(result)
+        return result
     elif not is_mask:
         return scale(extent.input, extent.initial, extent.initial_scaling, w, image, models)
     else: