mirror of
https://github.com/meta-llama/llama-stack.git
synced 2025-12-08 19:10:56 +00:00
mvp
This commit is contained in:
parent
4f07aca309
commit
3cbe3a72e8
10 changed files with 230 additions and 76 deletions
|
|
@ -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")
|
||||
|
|
|
|||
|
|
@ -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):
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue