298977887/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2025-01-11 02:15:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

Rename add_noise_mask -> noise_mask.

Browse Source
This commit is contained in:
comfyanonymous 2024-11-19 15:31:09 -05:00
parent 156a28786b
commit 898615122f
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
nodes.py
View File

@ -382,7 +382,7 @@ class InpaintModelConditioning:
"vae": ("VAE", ), "vae": ("VAE", ),
"pixels": ("IMAGE", ), "pixels": ("IMAGE", ),
"mask": ("MASK", ), "mask": ("MASK", ),
"add_noise_mask": ("BOOLEAN", {"default": True, "tooltip": "Add a noise mask to the latent so sampling will only happen within the mask. Might improve results or completely break things depending on the model."}), "noise_mask": ("BOOLEAN", {"default": True, "tooltip": "Add a noise mask to the latent so sampling will only happen within the mask. Might improve results or completely break things depending on the model."}),
}} }}
RETURN_TYPES = ("CONDITIONING","CONDITIONING","LATENT") RETURN_TYPES = ("CONDITIONING","CONDITIONING","LATENT")
@ -391,7 +391,7 @@ class InpaintModelConditioning:
CATEGORY = "conditioning/inpaint" CATEGORY = "conditioning/inpaint"
def encode(self, positive, negative, pixels, vae, mask, add_noise_mask): def encode(self, positive, negative, pixels, vae, mask, noise_mask):
x = (pixels.shape[1] // 8) * 8 x = (pixels.shape[1] // 8) * 8
y = (pixels.shape[2] // 8) * 8 y = (pixels.shape[2] // 8) * 8
mask = torch.nn.functional.interpolate(mask.reshape((-1, 1, mask.shape[-2], mask.shape[-1])), size=(pixels.shape[1], pixels.shape[2]), mode="bilinear") mask = torch.nn.functional.interpolate(mask.reshape((-1, 1, mask.shape[-2], mask.shape[-1])), size=(pixels.shape[1], pixels.shape[2]), mode="bilinear")
@ -415,7 +415,7 @@ class InpaintModelConditioning:
out_latent = {} out_latent = {}
out_latent["samples"] = orig_latent out_latent["samples"] = orig_latent
if add_noise_mask: if noise_mask:
out_latent["noise_mask"] = mask out_latent["noise_mask"] = mask
out = [] out = []