Skip to content

Commit be1d4a1

Browse files
authored
scripts : fix compare-llama-bench.py show parameter (#13514)
1 parent ab3971f commit be1d4a1

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

scripts/compare-llama-bench.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -113,7 +113,7 @@
113113
help_s = (
114114
"Columns to add to the table. "
115115
"Accepts a comma-separated list of values. "
116-
f"Legal values: {', '.join(KEY_PROPERTIES[:-2])}. "
116+
f"Legal values: {', '.join(KEY_PROPERTIES[:-3])}. "
117117
"Defaults to model name (model_type) and CPU and/or GPU name (cpu_info, gpu_info) "
118118
"plus any column where not all data points are the same. "
119119
"If the columns are manually specified, then the results for each unique combination of the "
@@ -505,7 +505,7 @@ def valid_format(data_files: list[str]) -> bool:
505505
show = known_args.show.split(",")
506506
unknown_cols = []
507507
for prop in show:
508-
if prop not in KEY_PROPERTIES[:-2]: # Last two values are n_prompt, n_gen.
508+
if prop not in KEY_PROPERTIES[:-3]: # Last three values are n_prompt, n_gen, n_depth.
509509
unknown_cols.append(prop)
510510
if unknown_cols:
511511
logger.error(f"Unknown values for --show: {', '.join(unknown_cols)}")

0 commit comments

Comments
 (0)