From 64273169b9a9a146a8dd354e53b463e514d4cb8f Mon Sep 17 00:00:00 2001 From: Vladimir Mandic Date: Sat, 5 Aug 2023 13:43:30 +0000 Subject: [PATCH] fix diffusers inpaint --- modules/img2img.py | 3 ++- modules/processing_diffusers.py | 2 +- 2 files changed, 3 insertions(+), 2 deletions(-) diff --git a/modules/img2img.py b/modules/img2img.py index 7ec50aaab..18a7aba85 100644 --- a/modules/img2img.py +++ b/modules/img2img.py @@ -106,7 +106,8 @@ def img2img(id_task: str, mode: int, prompt: str, negative_prompt: str, prompt_s elif mode == 2: # inpaint if init_img_with_mask is None: return - image, mask = init_img_with_mask["image"], init_img_with_mask["mask"] + image = init_img_with_mask["image"] + mask = init_img_with_mask["mask"] alpha_mask = ImageOps.invert(image.split()[-1]).convert('L').point(lambda x: 255 if x > 0 else 0, mode='1') mask = ImageChops.lighter(alpha_mask, mask.convert('L')).convert('L') image = image.convert("RGB") diff --git a/modules/processing_diffusers.py b/modules/processing_diffusers.py index 18c86116b..bb270b887 100644 --- a/modules/processing_diffusers.py +++ b/modules/processing_diffusers.py @@ -125,7 +125,7 @@ def process_diffusers(p: StableDiffusionProcessing, seeds, prompts, negative_pro task_specific_kwargs = {"image": p.init_images, "strength": p.denoising_strength} elif sd_models.get_diffusers_task(shared.sd_model) == sd_models.DiffusersTaskType.INPAINTING: p.ops.append('inpaint') - task_specific_kwargs = {"image": p.init_images, "mask_image": p.mask, "strength": p.denoising_strength} + task_specific_kwargs = {"image": p.init_images, "mask_image": p.mask, "strength": p.denoising_strength, "height": p.height, "width": p.width} # TODO diffusers use transformers for prompt parsing # from modules.prompt_parser import parse_prompt_attention