Skip to content

Commit bf0fdbd

Browse files
dunkeronihipsterusername
authored andcommitted
Fix: inpaint model mask using wrong tensor name
1 parent 731d317 commit bf0fdbd

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

invokeai/app/invocations/create_gradient_mask.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -184,7 +184,7 @@ def invoke(self, context: InvocationContext) -> GradientMaskOutput:
184184
main_model_config = context.models.get_config(self.unet.unet.key)
185185
assert isinstance(main_model_config, MainConfigBase)
186186
if main_model_config.variant is ModelVariantType.Inpaint:
187-
mask = mask_tensor
187+
mask = dilated_mask_tensor
188188
vae_info: LoadedModel = context.models.load(self.vae.vae)
189189
image = context.images.get_pil(self.image.image_name)
190190
image_tensor = image_resized_to_grid_as_tensor(image.convert("RGB"))

0 commit comments

Comments
 (0)