From fdf9b1c40c243d6806e787877e42ae4c115790ee Mon Sep 17 00:00:00 2001
From: Lincoln Stein <lincoln.stein@gmail.com>
Date: Sat, 29 Oct 2022 11:47:06 -0400
Subject: [PATCH] fix CLI inpainting crash

---
 ldm/generate.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/ldm/generate.py b/ldm/generate.py
index 4f8f486524..7f7bd43397 100644
--- a/ldm/generate.py
+++ b/ldm/generate.py
@@ -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