Update benchmark.py

LambdaLabsML · Jan 12, 2023 · df83334 · df83334
1 parent cdddab8
commit df83334
Showing 1 changed file with 2 additions and 1 deletion.
diff --git 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)
+    run_benchmark_grid(grid, n_repeats=args.repeats, num_inference_steps=args.steps, csv_fpath=args.output_file)