benchmarks: add diffusers SDXL

fal-ai · Nov 4, 2023 · ef9aa2a · ef9aa2a
1 parent 49c6566
commit ef9aa2a
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 2 deletions.
diff --git a/README.md b/README.md
@@ -7,8 +7,13 @@ better understanding of their performance and scalability.
 
 Running on an A100 80G SXM hosted at [fal.ai](https://fal.ai).
 
+Generation options:
+- `prompt="A photo of a cat"`
+- `num_inference_steps=50`
+
 <!-- START TABLE -->
 |                  | mean (s) | median (s) | min (s) | max (s) |
 |------------------|----------|------------|---------|---------|
-| Diffusers SD1.5  |   1.592s |     1.588s |  1.584s |  1.605s |
+| Diffusers SD1.5  |   1.619s |     1.614s |  1.603s |  1.645s |
+| Diffusers SDXL   |   5.916s |     5.917s |  5.909s |  5.927s |
 <!-- END TABLE -->
diff --git a/benchmarks/diffusers.py b/benchmarks/diffusers.py
@@ -44,5 +44,12 @@ def diffusers_any(
         "kwargs": {
             "model_name": "runwayml/stable-diffusion-v1-5",
         },
-    }
+    },
+    {
+        "name": "Diffusers SDXL",
+        "function": diffusers_any,
+        "kwargs": {
+            "model_name": "stabilityai/stable-diffusion-xl-base-1.0",
+        },
+    },
 ]