Removed **sentence-transformers/distilbert-base-nli-mean-tokens** as …

…default model and added **BAAI/bge-base-en-v1.5** as default model when no model param is given. (#3323) * Update hf.ts Removed sentence-transformers/distilbert-base-nli-mean-tokens as default model and added BAAI/bge-base-en-v1.5 as default model when no model param is given * Format --------- Co-authored-by: jacoblee93 <[email protected]>
langchain-ai · Nov 20, 2023 · d24872c · d24872c
1 parent 84fd5f2
commit d24872c
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/langchain/src/embeddings/hf.ts b/langchain/src/embeddings/hf.ts
@@ -32,8 +32,7 @@ export class HuggingFaceInferenceEmbeddings
   constructor(fields?: HuggingFaceInferenceEmbeddingsParams) {
     super(fields ?? {});
 
-    this.model =
-      fields?.model ?? "sentence-transformers/distilbert-base-nli-mean-tokens";
+    this.model = fields?.model ?? "BAAI/bge-base-en-v1.5";
     this.apiKey =
       fields?.apiKey ?? getEnvironmentVariable("HUGGINGFACEHUB_API_KEY");
     this.endpointUrl = fields?.endpointUrl;