From 1c73ac31ac6fdbc790e5e5eeffaf3ee900bf3bf3 Mon Sep 17 00:00:00 2001
From: zhudongwork <540953132@qq.com>
Date: Thu, 7 Nov 2024 03:45:05 +0000
Subject: [PATCH] feat:add bce-embedding-base_v1

---
 FlagEmbedding/inference/embedder/model_mapping.py | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/FlagEmbedding/inference/embedder/model_mapping.py b/FlagEmbedding/inference/embedder/model_mapping.py
index 582b4e9..c17436b 100644
--- a/FlagEmbedding/inference/embedder/model_mapping.py
+++ b/FlagEmbedding/inference/embedder/model_mapping.py
@@ -205,6 +205,11 @@ AUTO_EMBEDDER_MAPPING = OrderedDict([
     (
         'Linq-Embed-Mistral',
         EmbedderConfig(FlagLLMModel, PoolingMethod.LAST_TOKEN, query_instruction_format="Instruct: {}\nQuery: {}")
-    )
+    ),
+    # ============================== BCE ==============================
+    (
+        'bce-embedding-base_v1',
+        EmbedderConfig(FlagModel, PoolingMethod.CLS)
+    ),
     # TODO: Add more models, such as Jina, Stella_v5, NV-Embed, etc.
 ])