Merge branch 'fix-LoadImageMask' of https://github.com/flyingshutter/ComfyUI

f15b4a7f · comfyanonymous · 6c69853a · 9d1c610b · f15b4a7f
Commit f15b4a7f authored Apr 12, 2023 by comfyanonymous
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
@@ -940,6 +940,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():