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

Prevent indexing error for mode RGB

Browse Source 
I have not explicitly tested mode P
This commit is contained in:
spezialspezial 2022-10-29 23:21:53 +02:00 committed by Lincoln Stein
parent e4a212dfca
commit c2fab45a6e
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
ldm/generate.py
View File

@ -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)