github/InvokeAI
1
1
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI.git synced 2026-04-23 03:00:31 -04:00
Code Issues Packages Projects Releases Wiki Activity

Fix: inpaint model mask using wrong tensor name

Browse Source
This commit is contained in:
dunkeroni
2025-06-05 11:00:45 -04:00
committed by Kent Keirsey
parent 731d317a42
commit bf0fdbd10e
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
invokeai/app/invocations/create_gradient_mask.py
View File

@@ -184,7 +184,7 @@ class CreateGradientMaskInvocation(BaseInvocation):
main_model_config = context.models.get_config(self.unet.unet.key)
assert isinstance(main_model_config, MainConfigBase)
if main_model_config.variant is ModelVariantType.Inpaint:
mask = mask_tensor
mask = dilated_mask_tensor
vae_info: LoadedModel = context.models.load(self.vae.vae)
image = context.images.get_pil(self.image.image_name)
image_tensor = image_resized_to_grid_as_tensor(image.convert("RGB"))