[Fix] Hotfix for Subjective Evaluation (#686)

This commit is contained in:
bittersweet1999 2023-12-12 09:22:08 +08:00 committed by GitHub
parent 465308e430
commit 3e77175720
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 8 additions and 6 deletions

View File

@ -49,7 +49,7 @@ _meta_template = dict(
) )
judge_model = dict( judge_model = dict(
type=HuggingFaceCausalLM, type=HuggingFaceCausalLM,
abbr='qwen-7b-chat-hf', abbr='qwen-7b-chat-hf',
path="Qwen/Qwen-7B-Chat", path="Qwen/Qwen-7B-Chat",
@ -85,6 +85,7 @@ eval = dict(
max_num_workers=256, max_num_workers=256,
task=dict( task=dict(
type=SubjectiveEvalTask, type=SubjectiveEvalTask,
judge_cfg=judge_model
)), )),
) )
work_dir = './creation/' work_dir = './creation/'

View File

@ -102,8 +102,9 @@ class SubjectiveNaivePartitioner(NaivePartitioner):
else: else:
models = self.get_model_combinations(models, base_models, models = self.get_model_combinations(models, base_models,
compare_models) compare_models)
return super().partition(models=models, model_dataset_combinations = [{'models': models, 'datasets': datasets}]
datasets=datasets, return super().partition(
work_dir=work_dir, model_dataset_combinations=model_dataset_combinations,
out_dir=out_dir, work_dir=work_dir,
add_cfg=add_cfg) out_dir=out_dir,
add_cfg=add_cfg)