From df833345674409215c280ea6d74b92f4bdb7a7df Mon Sep 17 00:00:00 2001
From: Justin <605492+justinpinkney@users.noreply.github.com>
Date: Thu, 12 Jan 2023 11:09:45 +0000
Subject: [PATCH] Update benchmark.py

---
 scripts/benchmark.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

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)