fix CLI inpainting crash

This commit is contained in:
Lincoln Stein 2022-10-29 11:47:06 -04:00
parent 104466f5c0
commit fdf9b1c40c

View File

@ -947,7 +947,7 @@ class Generate:
def _create_init_image(self, image: Image.Image, width, height, fit=True):
if image.mode != 'RGBA':
image = image.convert('RGB')
image = image.convert('RGBA')
image = self._fit_image(image, (width, height)) if fit else self._squeeze_image(image)
return image