runner: Load truncated images

livepeer · Feb 2, 2024 · 823caea · 823caea
1 parent 21b607e
commit 823caea
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/runner/app/pipelines/image_to_image.py b/runner/app/pipelines/image_to_image.py
@@ -8,6 +8,10 @@
 from typing import List
 import logging
 
+from PIL import ImageFile
+
+ImageFile.LOAD_TRUNCATED_IMAGES = True
+
 logger = logging.getLogger(__name__)
 
 

diff --git a/runner/app/pipelines/image_to_video.py b/runner/app/pipelines/image_to_video.py
@@ -8,6 +8,10 @@
 from typing import List
 import logging
 
+from PIL import ImageFile
+
+ImageFile.LOAD_TRUNCATED_IMAGES = True
+
 logger = logging.getLogger(__name__)