Properly load higher bit depth videos. (#13542)

This commit is contained in:
comfyanonymous 2026-04-24 13:46:10 -07:00 committed by GitHub
parent 7636599389
commit 4304c15e9b
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -248,8 +248,8 @@ class VideoFromFile(VideoInput):
continue
if self.__duration and frame.pts >= end_pts:
break
img = frame.to_ndarray(format='rgb24') # shape: (H, W, 3)
img = torch.from_numpy(img) / 255.0 # shape: (H, W, 3)
img = frame.to_ndarray(format='gbrpf32le') # shape: (H, W, 3)
img = torch.from_numpy(img)
frames.append(img)
images = torch.stack(frames) if len(frames) > 0 else torch.zeros(0, 3, 0, 0)