Unverified Commit 5a05f266 authored by Eugene Khvedchenya's avatar Eugene Khvedchenya Committed by GitHub
Browse files

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


Signed-off-by: default avatarEugene Khvedchenia <ekhvedchenia@nvidia.com>
parent 7ef40bb9
...@@ -208,7 +208,7 @@ def video_to_pixel_values( ...@@ -208,7 +208,7 @@ def video_to_pixel_values(
) )
# dynamic_preprocess returns tensors already; take the single tile # dynamic_preprocess returns tensors already; take the single tile
assert len(pil_frame) >= 1 assert len(pil_frame) >= 1
frames_tensors.append(pil_frame[0]) frames_tensors.append(pil_frame[-1])
return torch.stack(frames_tensors) return torch.stack(frames_tensors)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment