Skip to content

Commit 2b5da79

Browse files
committed
missing csv header 'runtime'
1 parent 645fc33 commit 2b5da79

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

scripts/benchmark.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -137,7 +137,7 @@ def run_benchmark_grid(grid, n_repeats, num_inference_steps):
137137
csv_fpath = pathlib.Path(__file__).parent.parent / "benchmark_tmp.csv"
138138
# create benchmark.csv if not exists
139139
if not os.path.isfile(csv_fpath):
140-
header = ["device", "precision", "n_samples", "latency", "memory"]
140+
header = ["device", "precision", "runtime", "n_samples", "latency", "memory"]
141141
with open(csv_fpath, "w") as f:
142142
writer = csv.writer(f)
143143
writer.writerow(header)

0 commit comments

Comments
 (0)