fix wrong height and width when read video use torchvision (#41091)

2025-10-20 17:13:56 +08:00 · 2025-09-23 20:35:44 +08:00
parent 870add3daf
commit 946e5f95ea
1 changed files with 2 additions and 2 deletions
--- a/src/transformers/video_utils.py
+++ b/src/transformers/video_utils.py
@ -546,8 +546,8 @@ def read_video_torchvision(
    metadata.update(
        {
            "frames_indices": indices,
-            "height": video.shape[1],
-            "width": video.shape[2],
+            "height": video.shape[2],
+            "width": video.shape[3],
        }
    )
    return video, metadata