Mirrored_Repos/InvokeAI
1
0
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI synced 2024-08-30 20:32:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

Bug fix for inpaint size

Browse Source
This commit is contained in:
psychedelicious 2022-11-24 16:25:51 +11:00 committed by blessedcoolant
parent 9fe9301762
commit b93336dbf9
1 changed files with 7 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
ldm/invoke/generator/inpaint.py
View File

@ -189,6 +189,9 @@ class Inpaint(Img2Img):
self.enable_image_debugging = enable_image_debugging self.enable_image_debugging = enable_image_debugging
self.inpaint_width = inpaint_width
self.inpaint_height = inpaint_height
if isinstance(init_image, PIL.Image.Image): if isinstance(init_image, PIL.Image.Image):
self.pil_image = init_image.copy() self.pil_image = init_image.copy()
@ -292,10 +295,6 @@ class Inpaint(Img2Img):
result = self.sample_to_image(samples) 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) # Seam paint if this is our first pass (seam_size set to 0 during seam painting)
if seam_size > 0: if seam_size > 0:
old_image = self.pil_image or init_image old_image = self.pil_image or init_image
@ -335,6 +334,10 @@ class Inpaint(Img2Img):
gen_result = super().sample_to_image(samples).convert('RGB') gen_result = super().sample_to_image(samples).convert('RGB')
debug_image(gen_result, "gen_result", debug_status=self.enable_image_debugging) 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: if self.pil_image is None or self.pil_mask is None:
return gen_result return gen_result