Skip to content

Commit 068e86f

Browse files
authored
Merge pull request #1198 from zhudongwork/dev
feat:add bce-embedding-base_v1
2 parents e37bc46 + 1c73ac3 commit 068e86f

1 file changed

Lines changed: 6 additions & 1 deletion

File tree

FlagEmbedding/inference/embedder/model_mapping.py

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -205,6 +205,11 @@ class EmbedderConfig:
205205
(
206206
'Linq-Embed-Mistral',
207207
EmbedderConfig(FlagLLMModel, PoolingMethod.LAST_TOKEN, query_instruction_format="Instruct: {}\nQuery: {}")
208-
)
208+
),
209+
# ============================== BCE ==============================
210+
(
211+
'bce-embedding-base_v1',
212+
EmbedderConfig(FlagModel, PoolingMethod.CLS)
213+
),
209214
# TODO: Add more models, such as Jina, Stella_v5, NV-Embed, etc.
210215
])

0 commit comments

Comments
 (0)