make LoadImagesMask work with non RGBA images

9d1c610b · flyingshutter · 0bb5f93b · 9d1c610b
Commit 9d1c610b authored Apr 06, 2023 by flyingshutter
Show whitespace changes
Inline Side-by-side

Showing with 2 additions and 0 deletions

nodes.py nodes.py +2 -0

No files found.
--- a/nodes.py
+++ b/nodes.py
@@ -915,6 +915,8 @@ class LoadImageMask:
        input_dir = folder_paths.get_input_directory()
        image_path = os.path.join(input_dir, image)
        i = Image.open(image_path)
+        if i.getbands() != ("R", "G", "B", "A"):
+            i = i.convert("RGBA")
        mask = None
        c = channel[0].upper()
        if c in i.getbands():