From ae36a651af71127676363c1f618820025bda2b04 Mon Sep 17 00:00:00 2001 From: Dillon Laird Date: Tue, 20 Feb 2024 12:23:44 -0800 Subject: [PATCH] change default embedder --- lmm_tools/emb/emb.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/lmm_tools/emb/emb.py b/lmm_tools/emb/emb.py index 25696bc5..6186ddb4 100644 --- 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)