Load other jpeg formats without taking so much memory. (#13642)

This commit is contained in:
comfyanonymous 2026-04-30 16:33:09 -07:00 committed by GitHub
parent b633244635
commit e6e0936128
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -290,7 +290,7 @@ class VideoFromFile(VideoInput):
alphas = [] alphas = []
alpha_channel = True alpha_channel = True
break break
if frame.format.name in ("yuvj420p", "rgb24", "rgba", "pal8"): if frame.format.name in ("yuvj420p", "yuvj422p", "yuvj444p", "rgb24", "rgba", "pal8"):
process_image_format = lambda a: a.float() / 255.0 process_image_format = lambda a: a.float() / 255.0
if alpha_channel: if alpha_channel:
image_format = 'rgba' image_format = 'rgba'