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")

View file

@ -109,7 +109,7 @@ async def run_main(host: str, port: int, stream: bool):
cprint(f"User>{message.content}", "green")
iterator = client.chat_completion(
model="Llama3.1-8B-Instruct",
messages=[message],
messages=[message, UserMessage(content="write me 3 sentence about the sun.")],
stream=stream,
)
async for log in EventLogger().log(iterator):