Merge pull request #789 from EvolvingLMMs-Lab/fix/frames

fix(qwen2_5_vl): Ensure unique frame indices for videos with few frames

Author: Luodian

lmms_eval/models/simple/qwen2_5_vl.py

@@ -287,9 +287,12 @@ def _collate(x):
             if video_inputs is not None:
                 total_frames = video_inputs[0].shape[0]
                 indices = np.linspace(0, total_frames - 1, self.max_num_frames, dtype=int)
+                # Ensure unique indices if linspace produces duplicates for few frames
+                indices = np.unique(indices)
                 # Append the last frame index if not already included
                 if total_frames - 1 not in indices:
                     indices = np.append(indices, total_frames - 1)
+                    indices = np.unique(indices)  # Ensure uniqueness again
                 video_inputs[0] = video_inputs[0][indices]
             inputs = self.processor(text=texts, images=image_inputs, videos=video_inputs, padding=True, return_tensors="pt")