Unverified Commit ebe7226e authored by Baber Abbasi's avatar Baber Abbasi Committed by GitHub
Browse files

chat template hotfix (#2250)

* chat template hotfix

* pre-commit
parent aab42ba8
...@@ -287,12 +287,18 @@ def simple_evaluate( ...@@ -287,12 +287,18 @@ def simple_evaluate(
if check_integrity: if check_integrity:
run_task_tests(task_list=tasks) run_task_tests(task_list=tasks)
# hotfix: delete when chat_template fixed
try:
chat = lm.chat_template(apply_chat_template)
except: # noqa: E722
chat = None
if evaluation_tracker is not None: if evaluation_tracker is not None:
evaluation_tracker.general_config_tracker.log_experiment_args( evaluation_tracker.general_config_tracker.log_experiment_args(
model_source=model, model_source=model,
model_args=model_args, model_args=model_args,
system_instruction=system_instruction, system_instruction=system_instruction,
chat_template=lm.chat_template(apply_chat_template), chat_template=chat,
fewshot_as_multiturn=fewshot_as_multiturn, fewshot_as_multiturn=fewshot_as_multiturn,
) )
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment