From c33412288fbdcd132265c9029a38001fd9696aa5 Mon Sep 17 00:00:00 2001
From: comfyanonymous <comfyanonymous@protonmail.com>
Date: Tue, 7 May 2024 05:41:06 -0400
Subject: [PATCH] Fix issue with loading some JPG: #3416

---
 nodes.py | 14 ++++++++++++--
 1 file changed, 12 insertions(+), 2 deletions(-)

diff --git a/nodes.py b/nodes.py
index aa6d6fa9..4d3171b8 100644
--- a/nodes.py
+++ b/nodes.py
@@ -10,7 +10,7 @@ import time
 import random
 import logging
 
-from PIL import Image, ImageOps, ImageSequence
+from PIL import Image, ImageOps, ImageSequence, ImageFile
 from PIL.PngImagePlugin import PngInfo
 
 import numpy as np
@@ -1462,7 +1462,17 @@ class LoadImage:
         output_images = []
         output_masks = []
         for i in ImageSequence.Iterator(img):
-            i = ImageOps.exif_transpose(i)
+            prev_value = None
+            try:
+                i = ImageOps.exif_transpose(i)
+            except OSError:
+                prev_value = ImageFile.LOAD_TRUNCATED_IMAGES
+                ImageFile.LOAD_TRUNCATED_IMAGES = True
+                i = ImageOps.exif_transpose(i)
+            finally:
+                if prev_value is not None:
+                    ImageFile.LOAD_TRUNCATED_IMAGES = prev_value
+
             if i.mode == 'I':
                 i = i.point(lambda i: i * (1 / 255))
             image = i.convert("RGB")