@@ -433,7 +433,7 @@ def test_apply_chat_template_video_frame_sampling(self):
433433 num_frames = num_frames ,
434434 )
435435 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 )
437437
438438 # Load with `video_fps` arg
439439 video_fps = 1
@@ -445,7 +445,7 @@ def test_apply_chat_template_video_frame_sampling(self):
445445 video_fps = video_fps ,
446446 )
447447 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 )
449449
450450 # Load with `video_fps` and `num_frames` args, should raise an error
451451 with self .assertRaises (ValueError ):
@@ -466,7 +466,7 @@ def test_apply_chat_template_video_frame_sampling(self):
466466 return_dict = True ,
467467 )
468468 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 )
470470
471471 # Load video as a list of frames (i.e. images). NOTE: each frame should have same size
472472 # because we assume they come from one video
@@ -484,7 +484,7 @@ def test_apply_chat_template_video_frame_sampling(self):
484484 return_dict = True ,
485485 )
486486 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 )
488488
489489 @require_av
490490 def test_apply_chat_template_video_special_processing (self ):
0 commit comments