yujunjun/FlagEmbedding
1
0
Fork 0
mirror of https://github.com/FlagOpen/FlagEmbedding.git synced 2026-01-10 14:20:28 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat:add bce-embedding-base_v1

Browse Source
This commit is contained in:
zhudongwork 2024-11-07 03:45:05 +00:00
parent e37bc46137
commit 1c73ac31ac
1 changed files with 6 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
FlagEmbedding/inference/embedder/model_mapping.py
View File

@ -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.
])