eleuther eval fix

This commit is contained in:
Xi Yan 2024-10-11 09:30:10 -07:00
parent ad18dc94ac
commit fb565dfb06
3 changed files with 24 additions and 20 deletions

View file

@ -54,27 +54,29 @@ class EvaluationClient(Evals):
async def run_main(host: str, port: int):
client = EvaluationClient(f"http://{host}:{port}")
# CustomDataset
# Custom Eval Task
# response = await client.run_evals(
# model="Llama3.1-8B-Instruct",
# dataset="mmlu-simple-eval-en",
# task="mmlu",
# eval_task_config=EvaluateTaskConfig(
# n_samples=2,
# ),
# )
# Eleuther Eval Task
response = await client.run_evals(
model="Llama3.1-8B-Instruct",
dataset="mmlu-simple-eval-en",
task="mmlu",
# task="meta_mmlu_pro_instruct",
task="meta_ifeval",
eval_task_config=EvaluateTaskConfig(
n_samples=2,
),
)
cprint(f"evaluate response={response}", "green")
# Eleuther Eval Task
# response = await client.run_evals(
# model="Llama3.1-8B-Instruct",
# task="meta_mmlu_pro_instruct",
# # task="meta_ifeval",
# eval_task_config=EvaluateTaskConfig(
# n_samples=2,
# )
# )
# cprint(response.metrics["metrics_table"], "red")
if response.formatted_report:
cprint(response.formatted_report, "green")
else:
cprint(f"evaluate response={response}", "green")
def main(host: str, port: int):

View file

@ -157,12 +157,14 @@ class EleutherEvalsAdapter(Evals):
limit=eval_task_config.n_samples,
)
eval_result = EvalResult(
metrics={},
)
formatted_output = lm_eval.utils.make_table(output)
cprint(formatted_output, "green")
return EvaluateResponse(
metrics={
"metrics_table": formatted_output,
},
eval_result=eval_result,
formatted_report=formatted_output,
)

View file

@ -14,8 +14,8 @@ apis:
- evals
providers:
evals:
- provider_id: meta-reference
provider_type: meta-reference
- provider_id: eleuther
provider_type: eleuther
config: {}
inference:
- provider_id: remote::tgi