mirror of
https://github.com/vllm-project/vllm.git
synced 2025-10-20 14:53:52 +08:00
@ -692,6 +692,8 @@ def main(args: argparse.Namespace):
|
||||
]:
|
||||
if field in result_json:
|
||||
del result_json[field]
|
||||
if field in benchmark_result:
|
||||
del benchmark_result[field]
|
||||
|
||||
# Traffic
|
||||
result_json["request_rate"] = (args.request_rate if args.request_rate
|
||||
@ -711,7 +713,10 @@ def main(args: argparse.Namespace):
|
||||
file_name = args.result_filename
|
||||
if args.result_dir:
|
||||
file_name = os.path.join(args.result_dir, file_name)
|
||||
with open(file_name, "w", encoding='utf-8') as outfile:
|
||||
with open(file_name, mode="a+", encoding='utf-8') as outfile:
|
||||
# Append a newline.
|
||||
if outfile.tell() != 0:
|
||||
outfile.write("\n")
|
||||
json.dump(result_json, outfile)
|
||||
save_to_pytorch_benchmark_format(args, result_json, file_name)
|
||||
|
||||
|
7
benchmarks/convert_to_csv.py
Normal file
7
benchmarks/convert_to_csv.py
Normal file
@ -0,0 +1,7 @@
|
||||
import pandas as pd
|
||||
|
||||
df_sgl = pd.read_json("sgl-results.json", lines=True)
|
||||
df_sgl.to_csv("sgl-results.csv")
|
||||
|
||||
df_vllm = pd.read_json("vllm-results.json", lines=True)
|
||||
df_vllm.to_csv("vllm-results.csv")
|
Reference in New Issue
Block a user