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

merge outcropping and postprocessing fixes

Browse Source
This commit is contained in:
Lincoln Stein 2022-10-04 22:47:21 -04:00
parent 4c482fe24a
commit a09c64a1fe
1 changed files with 2 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
ldm/generate.py
View File

@ -624,9 +624,7 @@ class Generate:
img, img,
width, width,
height, height,
) # this returns an Image )
init_image = self._create_init_image(image) # this returns a torch tensor
# if image has a transparent area and no mask was provided, then try to generate mask # if image has a transparent area and no mask was provided, then try to generate mask
if self._has_transparency(image) and not mask: if self._has_transparency(image) and not mask:
@ -898,7 +896,7 @@ class Generate:
image = 2.0 * image - 1.0 image = 2.0 * image - 1.0
return image.to(self.device) return image.to(self.device)
def _create_init_mask(self, image): def _create_init_mask(self, image, width, height, fit=True):
# convert into a black/white mask # convert into a black/white mask
image = self._image_to_mask(image) image = self._image_to_mask(image)
image = image.convert('RGB') image = image.convert('RGB')