Support properly loading images with mode I.

6a10640f · comfyanonymous · c6951548 · 6a10640f
Commit 6a10640f authored Jan 08, 2024 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
@@ -1415,6 +1415,8 @@ class LoadImage:
        output_masks = []
        for i in ImageSequence.Iterator(img):
            i = ImageOps.exif_transpose(i)
+            if i.mode == 'I':
+                i = i.point(lambda i: i * (1 / 255))
            image = i.convert("RGB")
            image = np.array(image).astype(np.float32) / 255.0
            image = torch.from_numpy(image)[None,]