From eedf218edeb06789a8021b83fd6e05b9a25ee9ed Mon Sep 17 00:00:00 2001 From: sayakpaul Date: Tue, 5 Dec 2023 18:15:42 +0530 Subject: [PATCH] fix: mapping for resolution --- benchmarks/base_classes.py | 1 + 1 file changed, 1 insertion(+) diff --git a/benchmarks/base_classes.py b/benchmarks/base_classes.py index ca38b144de5b..194605f75fa5 100644 --- a/benchmarks/base_classes.py +++ b/benchmarks/base_classes.py @@ -38,6 +38,7 @@ "TencentARC/t2iadapter_canny_sd14v1": (512, 512), "TencentARC/t2i-adapter-canny-sdxl-1.0": (1024, 1024), "stabilityai/stable-diffusion-2-1": (768, 768), + "stabilityai/stable-diffusion-xl-base-1.0": (1024, 1024), "stabilityai/stable-diffusion-xl-refiner-1.0": (1024, 1024), }