From c2fab45a6e729cf8425f32da2debd70ac10de724 Mon Sep 17 00:00:00 2001
From: spezialspezial <75758219+spezialspezial@users.noreply.github.com>
Date: Sat, 29 Oct 2022 23:21:53 +0200
Subject: [PATCH] Prevent indexing error for mode RGB

I have not explicitly tested mode P
---
 ldm/generate.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/ldm/generate.py b/ldm/generate.py
index 7f7bd43397..6c45c02d47 100644
--- a/ldm/generate.py
+++ b/ldm/generate.py
@@ -965,6 +965,8 @@ class Generate:
         # Obtain the mask from the transparency channel
         if mask_image.mode == 'L':
             mask = mask_image
+        elif mask_image.mode in ('RGB', 'P'):
+            mask = mask_image.convert('L')
         else:
             # Obtain the mask from the transparency channel
             mask = Image.new(mode="L", size=mask_image.size, color=255)