When loading a model, the similarity_fn_name attribute defaults to cosine instead of max_sim. Since SentenceTransformer does not support max_sim natively, this functionality must be implemented using pylate.

self.model = colbert_model.ColBERT("colbert-ir/colbertv2.0")