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

2 years ago · f15b4a7ffa
1 changed files with 2 additions and 0 deletions
--- 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():