BUG FIX: the model wasn't able to call sample frames

#12
Files changed (1) hide show
  1. scripts/qwen3_vl_reranker.py +1 -1
scripts/qwen3_vl_reranker.py CHANGED
@@ -201,7 +201,7 @@ class Qwen3VLReranker():
201
  if isinstance(video, list):
202
  video_content = video
203
  if self.num_frames is not None or self.max_frames is not None:
204
- video_content = self._sample_frames(video_content, self.num_frames, self.max_frames)
205
  video_content = [
206
  ('file://' + ele if isinstance(ele, str) else ele)
207
  for ele in video_content
 
201
  if isinstance(video, list):
202
  video_content = video
203
  if self.num_frames is not None or self.max_frames is not None:
204
+ video_content = sample_frames(video_content, self.num_frames, self.max_frames)
205
  video_content = [
206
  ('file://' + ele if isinstance(ele, str) else ele)
207
  for ele in video_content