From b93336dbf98ab64c97bb1f0ec70920746d802952 Mon Sep 17 00:00:00 2001 From: psychedelicious <4822129+psychedelicious@users.noreply.github.com> Date: Thu, 24 Nov 2022 16:25:51 +1100 Subject: [PATCH] Bug fix for inpaint size --- ldm/invoke/generator/inpaint.py | 11 +++++++---- 1 file changed, 7 insertions(+), 4 deletions(-) diff --git a/ldm/invoke/generator/inpaint.py b/ldm/invoke/generator/inpaint.py index 2ff0ca248a..bf63223016 100644 --- a/ldm/invoke/generator/inpaint.py +++ b/ldm/invoke/generator/inpaint.py @@ -189,6 +189,9 @@ class Inpaint(Img2Img): self.enable_image_debugging = enable_image_debugging + self.inpaint_width = inpaint_width + self.inpaint_height = inpaint_height + if isinstance(init_image, PIL.Image.Image): self.pil_image = init_image.copy() @@ -292,10 +295,6 @@ class Inpaint(Img2Img): result = self.sample_to_image(samples) - # Resize if necessary - if inpaint_width and inpaint_height: - result = result.resize(self.pil_image.size) - # Seam paint if this is our first pass (seam_size set to 0 during seam painting) if seam_size > 0: old_image = self.pil_image or init_image @@ -335,6 +334,10 @@ class Inpaint(Img2Img): gen_result = super().sample_to_image(samples).convert('RGB') debug_image(gen_result, "gen_result", debug_status=self.enable_image_debugging) + # Resize if necessary + if self.inpaint_width and self.inpaint_height: + gen_result = gen_result.resize(self.pil_image.size) + if self.pil_image is None or self.pil_mask is None: return gen_result