@@ -433,7 +433,7 @@ def test_apply_chat_template_video_frame_sampling(self):
433
433
num_frames = num_frames ,
434
434
)
435
435
self .assertTrue (self .videos_input_name in out_dict_with_video )
436
- self .assertEqual (len (out_dict_with_video [self .videos_input_name ]), 9568 )
436
+ self .assertEqual (len (out_dict_with_video [self .videos_input_name ]), 5760 )
437
437
438
438
# Load with `video_fps` arg
439
439
video_fps = 1
@@ -445,7 +445,7 @@ def test_apply_chat_template_video_frame_sampling(self):
445
445
video_fps = video_fps ,
446
446
)
447
447
self .assertTrue (self .videos_input_name in out_dict_with_video )
448
- self .assertEqual (len (out_dict_with_video [self .videos_input_name ]), 23920 )
448
+ self .assertEqual (len (out_dict_with_video [self .videos_input_name ]), 14400 )
449
449
450
450
# Load with `video_fps` and `num_frames` args, should raise an error
451
451
with self .assertRaises (ValueError ):
@@ -466,7 +466,7 @@ def test_apply_chat_template_video_frame_sampling(self):
466
466
return_dict = True ,
467
467
)
468
468
self .assertTrue (self .videos_input_name in out_dict_with_video )
469
- self .assertEqual (len (out_dict_with_video [self .videos_input_name ]), 717600 )
469
+ self .assertEqual (len (out_dict_with_video [self .videos_input_name ]), 432000 )
470
470
471
471
# Load video as a list of frames (i.e. images). NOTE: each frame should have same size
472
472
# because we assume they come from one video
@@ -484,7 +484,7 @@ def test_apply_chat_template_video_frame_sampling(self):
484
484
return_dict = True ,
485
485
)
486
486
self .assertTrue (self .videos_input_name in out_dict_with_video )
487
- self .assertEqual (len (out_dict_with_video [self .videos_input_name ]), 5704 )
487
+ self .assertEqual (len (out_dict_with_video [self .videos_input_name ]), 2904 )
488
488
489
489
@require_av
490
490
def test_apply_chat_template_video_special_processing (self ):
0 commit comments