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,