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

Force RGB for img2img

Browse Source
This commit is contained in:
Kyle Schouviller 2022-10-26 12:09:38 -07:00
parent bd8bb8c80b
commit d05373d35a
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
ldm/invoke/generator/img2img.py
View File

@ -29,7 +29,7 @@ class Img2Img(Generator):
) )
if isinstance(init_image, PIL.Image.Image): if isinstance(init_image, PIL.Image.Image):
init_image = self._image_to_tensor(init_image) init_image = self._image_to_tensor(init_image.convert('RGB'))
scope = choose_autocast(self.precision) scope = choose_autocast(self.precision)
with scope(self.model.device.type): with scope(self.model.device.type):