change default embedder

landing-ai · Feb 20, 2024 · ae36a65 · ae36a65
1 parent c994011
commit ae36a65
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/lmm_tools/emb/emb.py b/lmm_tools/emb/emb.py
@@ -12,7 +12,7 @@ def embed(self, text: str) -> npt.NDArray[np.float32]:
 
 
 class SentenceTransformerEmb(Embedder):
- def __init__(self, model_name: str = "all-MiniLM-L12-v2"):
+ def __init__(self, model_name: str = "BAAI/bge-small-en-v1.5"):
  from sentence_transformers import SentenceTransformer
 
  self.model = SentenceTransformer(model_name)