Fix issue of using only the part of video frame [Nemotron Nano] (#26186)

Signed-off-by: Eugene Khvedchenia <ekhvedchenia@nvidia.com>
This commit is contained in:
Eugene Khvedchenya
2025-10-04 03:21:00 +03:00
committed by GitHub
parent 7ef40bb983
commit 5a05f26603

View File

@ -208,7 +208,7 @@ def video_to_pixel_values(
)
# dynamic_preprocess returns tensors already; take the single tile
assert len(pil_frame) >= 1
frames_tensors.append(pil_frame[0])
frames_tensors.append(pil_frame[-1])
return torch.stack(frames_tensors)