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)