diff --git a/mteb/models/model2vec_models.py b/mteb/models/model2vec_models.py index e0109be125..1541d3ca3d 100644 --- a/mteb/models/model2vec_models.py +++ b/mteb/models/model2vec_models.py @@ -50,7 +50,7 @@ def encode( Returns: The encoded sentences. """ - return self.static_model.encode(sentences) + return self.static_model.encode(sentences).astype(np.float32) m2v_base_glove_subword = ModelMeta(