wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-05-30 02:47:24 +08:00
Code Issues Actions 16 Packages Projects Releases Wiki Activity

fix(LoadImage): ignore broken frames after first decoded image

Browse Source
This commit is contained in:
bigcat88 2025-12-30 18:29:56 +02:00
parent d7111e426a
commit db60a1e0e9
No known key found for this signature in database
GPG Key ID: 1F0BF0EC3CF22721
1 changed files with 26 additions and 22 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
nodes.py
View File

@ -1665,6 +1665,7 @@ class LoadImage:
excluded_formats = ['MPO'] excluded_formats = ['MPO']
try:
for i in ImageSequence.Iterator(img): for i in ImageSequence.Iterator(img):
i = node_helpers.pillow(ImageOps.exif_transpose, i) i = node_helpers.pillow(ImageOps.exif_transpose, i)
@ -1691,6 +1692,9 @@ class LoadImage:
mask = torch.zeros((64,64), dtype=torch.float32, device="cpu") mask = torch.zeros((64,64), dtype=torch.float32, device="cpu")
output_images.append(image) output_images.append(image)
output_masks.append(mask.unsqueeze(0)) output_masks.append(mask.unsqueeze(0))
except ValueError:
if img.format != "MPO" or len(output_images) == 0:
raise
if len(output_images) > 1 and img.format not in excluded_formats: if len(output_images) > 1 and img.format not in excluded_formats:
output_image = torch.cat(output_images, dim=0) output_image = torch.cat(output_images, dim=0)