From 2ff964ee329a84852f9d8577eb209a821946e706 Mon Sep 17 00:00:00 2001 From: lnxtree Date: Thu, 26 Mar 2026 20:59:03 +0800 Subject: [PATCH] fix(eval): pass use_bf16 through AbsEvalRunner.get_model and COIR evaluator model initialization --- FlagEmbedding/abc/evaluation/runner.py | 1 + research/BGE_Coder/evaluation/coir_eval/main.py | 2 ++ 2 files changed, 3 insertions(+) diff --git a/FlagEmbedding/abc/evaluation/runner.py b/FlagEmbedding/abc/evaluation/runner.py index 2ab2195f..0a8a6df6 100644 --- a/FlagEmbedding/abc/evaluation/runner.py +++ b/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, diff --git a/research/BGE_Coder/evaluation/coir_eval/main.py b/research/BGE_Coder/evaluation/coir_eval/main.py index 3525b4b5..c6af8540 100644 --- a/research/BGE_Coder/evaluation/coir_eval/main.py +++ b/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,