Merge branch 'FlagOpen:master' into master

Author: ZiyiXia

FlagEmbedding/inference/embedder/model_mapping.py

@@ -205,6 +205,11 @@ class EmbedderConfig:
     (
         '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.
 ])

scripts/add_reranker_score.py

@@ -29,7 +29,7 @@ class ModelArgs:
         default=None, metadata={"help": "The reranker name or path."}
     )
     reranker_model_class: Optional[str] = field(
-        default="auto", metadata={"help": "The reranker model class. Available classes: ['auto', 'encoder-only-base', 'decoder-only-base', 'decoder-only-layerwise', 'decoder-only-lightweight']. Default: auto.", "choices": ["auto", "encoder-only-base", "decoder-only-base", "decoder-only-layerwise", "decoder-only-lightweight"]}
+        default=None, metadata={"help": "The reranker model class. Available classes: ['auto', 'encoder-only-base', 'decoder-only-base', 'decoder-only-layerwise', 'decoder-only-lightweight']. Default: auto.", "choices": ["encoder-only-base", "decoder-only-base", "decoder-only-layerwise", "decoder-only-lightweight"]}
     )
     reranker_peft_path: Optional[str] = field(
         default=None, metadata={"help": "The reranker peft path."}