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,
width,
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 self._has_transparency(image) and not mask:
@ -898,7 +896,7 @@ class Generate:
image = 2.0 * image - 1.0
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
image = self._image_to_mask(image)
image = image.convert('RGB')