Skip to content

GitLab

Unverified Commit 808a7b69 authored by lkchen's avatar lkchen Committed by GitHub
Browse files

[bench] Fix benchmark/serve.py to ignore unavailable results (#22382) 


Signed-off-by: default avatarLinkun <github@lkchen.net>
parent 099c0464
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
vllm/benchmarks/serve.py
View file @ 808a7b69
...@@ -665,7 +665,7 @@ def save_to_pytorch_benchmark_format(args: argparse.Namespace, ...@@ -665,7 +665,7 @@ def save_to_pytorch_benchmark_format(args: argparse.Namespace,
pt_records = convert_to_pytorch_benchmark_format( pt_records = convert_to_pytorch_benchmark_format(
args=args, args=args,
metrics={k: [results[k]] metrics={k: [results[k]]
for k in metrics}, for k in metrics if k in results},
extra_info={ extra_info={
k: results[k] k: results[k]
for k in results if k not in metrics and k not in ignored_metrics for k in results if k not in metrics and k not in ignored_metrics
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment