phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-12-08 19:10:56 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

move folder

Browse source
This commit is contained in:
Xi Yan 2024-10-05 11:57:21 -07:00
parent 6234dd97d5
commit 041634192a
9 changed files with 18 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

13
llama_stack/apis/evals/client.py
View file

@ -42,12 +42,21 @@ class EvaluationClient(Evals):
async def run_main(host: str, port: int):
client = EvaluationClient(f"http://{host}:{port}")
# CustomDataset
# response = await client.run_evals(
# "Llama3.1-8B-Instruct",
# "mmlu-simple-eval-en",
# "mmlu",
# )
# cprint(f"evaluate response={response}", "green")
# Eleuther Eval
response = await client.run_evals(
"Llama3.1-8B-Instruct",
"mmlu-simple-eval-en",
"PLACEHOLDER_DATASET_NAME",
"mmlu",
)
cprint(f"evaluate response={response}", "green")
cprint(response.metrics["metrics_table"], "red")
def main(host: str, port: int):

2
llama_stack/apis/inference/client.py
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, UserMessage(content="write me 3 sentence about the sun.")],
messages=[message],
stream=stream,
)
async for log in EventLogger().log(iterator):