[Minor] fix lint in main (#10128)
This commit is contained in:
@@ -85,7 +85,10 @@ class BenchArgs:
|
|||||||
)
|
)
|
||||||
parser.add_argument("--profile-by-stage", action="store_true")
|
parser.add_argument("--profile-by-stage", action="store_true")
|
||||||
parser.add_argument(
|
parser.add_argument(
|
||||||
"--dataset-path", type=str, default=BenchArgs.dataset_path, help="Path to the dataset."
|
"--dataset-path",
|
||||||
|
type=str,
|
||||||
|
default=BenchArgs.dataset_path,
|
||||||
|
help="Path to the dataset.",
|
||||||
)
|
)
|
||||||
|
|
||||||
@classmethod
|
@classmethod
|
||||||
@@ -350,7 +353,7 @@ def run_benchmark(server_args: ServerArgs, bench_args: BenchArgs):
|
|||||||
run_name="",
|
run_name="",
|
||||||
result_filename="",
|
result_filename="",
|
||||||
tokenizer=tokenizer,
|
tokenizer=tokenizer,
|
||||||
dataset_path=bench_args.dataset_path
|
dataset_path=bench_args.dataset_path,
|
||||||
)
|
)
|
||||||
print("=" * 8 + " Warmup End " + "=" * 8 + "\n")
|
print("=" * 8 + " Warmup End " + "=" * 8 + "\n")
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user