diff --git a/fastembed/late_interaction_multimodal/colpali.py b/fastembed/late_interaction_multimodal/colpali.py index c6b98e23..9fab9535 100644 --- a/fastembed/late_interaction_multimodal/colpali.py +++ b/fastembed/late_interaction_multimodal/colpali.py @@ -175,7 +175,7 @@ def _preprocess_onnx_text_input( ) -> dict[str, NumpyArray]: onnx_input["input_ids"] = np.array( [ - self.QUERY_MARKER_TOKEN_ID + input_ids[2:].tolist() + self.QUERY_MARKER_TOKEN_ID + input_ids[2:].tolist() # type: ignore[index] for input_ids in onnx_input["input_ids"] ] )