From ef9aa2a0315b54b13aac77f62dc9a086985f6ae9 Mon Sep 17 00:00:00 2001 From: Batuhan Taskaya Date: Sat, 4 Nov 2023 18:40:37 +0300 Subject: [PATCH] benchmarks: add diffusers SDXL --- README.md | 7 ++++++- benchmarks/diffusers.py | 9 ++++++++- 2 files changed, 14 insertions(+), 2 deletions(-) diff --git a/README.md b/README.md index d677629..7ef6ee0 100644 --- 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` + | | 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 | diff --git a/benchmarks/diffusers.py b/benchmarks/diffusers.py index 924aba2..d2b63c9 100644 --- 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", + }, + }, ]