fix: Cast all Model2Vec outputs as floats (#1667)

cast all outputs as floats
embeddings-benchmark · Jan 1, 2025 · fa0ed6b · fa0ed6b
1 parent e4edb66
commit fa0ed6b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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(