This commit is contained in:
Xi Yan 2024-10-04 00:25:57 -07:00
parent 4f07aca309
commit 3cbe3a72e8
10 changed files with 230 additions and 76 deletions

View file

@ -33,6 +33,7 @@ class EvaluationClient(Evals):
"task": task,
},
headers={"Content-Type": "application/json"},
timeout=3600,
)
response.raise_for_status()
return EvaluateResponse(**response.json())
@ -43,7 +44,7 @@ async def run_main(host: str, port: int):
response = await client.run_evals(
"Llama3.1-8B-Instruct",
"mmlu.csv",
"mmlu-simple-eval-en",
"mmlu",
)
cprint(f"evaluate response={response}", "green")