Merge pull request #1568 from lnxtree/feat-fix-embedder_use_bf16

fix(eval): pass use_bf16 through AbsEvalRunner.get_model and COIR evaluator model initialization

Author: hanhainebula

FlagEmbedding/abc/evaluation/runner.py

@@ -50,6 +50,7 @@ def get_models(model_args: AbsEvalModelArgs) -> Tuple[AbsEmbedder, Union[AbsRera
             normalize_embeddings=model_args.normalize_embeddings,
             pooling_method=model_args.pooling_method,
             use_fp16=model_args.use_fp16,
+            use_bf16=model_args.use_bf16,
             query_instruction_for_retrieval=model_args.query_instruction_for_retrieval,
             query_instruction_format=model_args.query_instruction_format_for_retrieval,
             devices=model_args.devices,

research/BGE_Coder/evaluation/coir_eval/main.py

@@ -17,6 +17,7 @@ def get_model(model_args: COIREvalModelArgs):
             normalize_embeddings=model_args.normalize_embeddings,
             pooling_method=model_args.pooling_method,
             use_fp16=model_args.use_fp16,
+            use_bf16=model_args.use_bf16,
             query_instruction_for_retrieval=model_args.query_instruction_for_retrieval,
             query_instruction_format=model_args.query_instruction_format_for_retrieval,
             devices=model_args.devices,
@@ -32,6 +33,7 @@ def get_model(model_args: COIREvalModelArgs):
             normalize_embeddings=model_args.normalize_embeddings,
             pooling_method=model_args.pooling_method,
             use_fp16=model_args.use_fp16,
+            use_bf16=model_args.use_bf16,
             query_instruction_for_retrieval=model_args.query_instruction_for_retrieval,
             query_instruction_format=model_args.query_instruction_format_for_retrieval,
             devices=model_args.devices,