diff --git a/scripts/benchmark.py b/scripts/benchmark.py index f7ce04c..eef4818 100644 --- a/scripts/benchmark.py +++ b/scripts/benchmark.py @@ -224,6 +224,7 @@ def run_benchmark_grid(grid, n_repeats, num_inference_steps, csv_fpath): device_desc, precision, use_autocast, + xformers, backend, n_samples, latency, @@ -290,4 +291,4 @@ def run_benchmark_grid(grid, n_repeats, num_inference_steps, csv_fpath): # https://github.com/huggingface/diffusers/pull/440 "backend": ("pytorch", "onnx") if device.type == "cpu" else ("pytorch",), } - run_benchmark_grid(grid, n_repeats=args.repeats, num_inference_steps=args.steps, csv_fpath=args.output_file) \ No newline at end of file + run_benchmark_grid(grid, n_repeats=args.repeats, num_inference_steps=args.steps, csv_fpath=args.output_file)