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:
Kyle Schouviller 2022-11-05 14:47:53 -07:00 committed by Lincoln Stein
parent 09e41e8f76
commit f91fd27624
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

@ -160,6 +160,9 @@ class Inpaint(Img2Img):
the time you call it. kwargs are 'init_latent' and 'strength' the time you call it. kwargs are 'init_latent' and 'strength'
""" """
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 self.pil_image = init_image
@ -253,10 +256,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:
result = self.seam_paint( result = self.seam_paint(
@ -323,6 +322,10 @@ class Inpaint(Img2Img):
def sample_to_image(self, samples)->Image.Image: def sample_to_image(self, samples)->Image.Image:
gen_result = super().sample_to_image(samples).convert('RGB') gen_result = super().sample_to_image(samples).convert('RGB')
# 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