Support properly loading images with mode I.

jn-jairo · Jan 8, 2024 · 6a10640 · 6a10640
1 parent c695154
commit 6a10640
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/nodes.py b/nodes.py
@@ -1415,6 +1415,8 @@ def load_image(self, image):
         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,]