Fix --dataset-path in bench_one_batch_server (#10475)

This commit is contained in:
Liangsheng Yin
2025-09-16 02:55:02 +08:00
committed by GitHub
parent 3b25dc127a
commit 2cf811a9da
2 changed files with 4 additions and 2 deletions

View File

@@ -377,6 +377,7 @@ def run_benchmark(server_args: ServerArgs, bench_args: BenchArgs):
run_name=bench_args.run_name,
result_filename=bench_args.result_filename,
tokenizer=tokenizer,
dataset_path=bench_args.dataset_path,
)
)
@@ -402,6 +403,7 @@ def run_benchmark(server_args: ServerArgs, bench_args: BenchArgs):
profile=bench_args.profile,
profile_steps=bench_args.profile_steps,
profile_by_stage=bench_args.profile_by_stage,
dataset_path=bench_args.dataset_path,
)[-1],
)
)