diff --git a/benchmarks/base_classes.py b/benchmarks/base_classes.py index dc1ca72388af..45bf65c93c93 100644 --- a/benchmarks/base_classes.py +++ b/benchmarks/base_classes.py @@ -34,7 +34,7 @@ RESOLUTION_MAPPING = { - "runwayml/stable-diffusion-v1-5": (512, 512), + "Lykon/DreamShaper": (512, 512), "lllyasviel/sd-controlnet-canny": (512, 512), "diffusers/controlnet-canny-sdxl-1.0": (1024, 1024), "TencentARC/t2iadapter_canny_sd14v1": (512, 512), @@ -268,7 +268,7 @@ def run_inference(self, pipe, args): class ControlNetBenchmark(TextToImageBenchmark): pipeline_class = StableDiffusionControlNetPipeline aux_network_class = ControlNetModel - root_ckpt = "runwayml/stable-diffusion-v1-5" + root_ckpt = "Lykon/DreamShaper" url = "https://huggingface.co/datasets/diffusers/docs-images/resolve/main/benchmarking/canny_image_condition.png" image = load_image(url).convert("RGB") @@ -311,7 +311,7 @@ def __init__(self, args): class T2IAdapterBenchmark(ControlNetBenchmark): pipeline_class = StableDiffusionAdapterPipeline aux_network_class = T2IAdapter - root_ckpt = "CompVis/stable-diffusion-v1-4" + root_ckpt = "Lykon/DreamShaper" url = "https://huggingface.co/datasets/diffusers/docs-images/resolve/main/benchmarking/canny_for_adapter.png" image = load_image(url).convert("L") diff --git a/benchmarks/benchmark_ip_adapters.py b/benchmarks/benchmark_ip_adapters.py index 5c11ab38383c..9a31a21fc60d 100644 --- a/benchmarks/benchmark_ip_adapters.py +++ b/benchmarks/benchmark_ip_adapters.py @@ -7,7 +7,8 @@ IP_ADAPTER_CKPTS = { - "runwayml/stable-diffusion-v1-5": ("h94/IP-Adapter", "ip-adapter_sd15.bin"), + # because original SD v1.5 has been taken down. + "Lykon/DreamShaper": ("h94/IP-Adapter", "ip-adapter_sd15.bin"), "stabilityai/stable-diffusion-xl-base-1.0": ("h94/IP-Adapter", "ip-adapter_sdxl.bin"), } @@ -17,7 +18,7 @@ parser.add_argument( "--ckpt", type=str, - default="runwayml/stable-diffusion-v1-5", + default="rstabilityai/stable-diffusion-xl-base-1.0", choices=list(IP_ADAPTER_CKPTS.keys()), ) parser.add_argument("--batch_size", type=int, default=1) diff --git a/benchmarks/benchmark_sd_img.py b/benchmarks/benchmark_sd_img.py index 491e7c9a65a9..772befe8795f 100644 --- a/benchmarks/benchmark_sd_img.py +++ b/benchmarks/benchmark_sd_img.py @@ -11,9 +11,9 @@ parser.add_argument( "--ckpt", type=str, - default="runwayml/stable-diffusion-v1-5", + default="Lykon/DreamShaper", choices=[ - "runwayml/stable-diffusion-v1-5", + "Lykon/DreamShaper", "stabilityai/stable-diffusion-2-1", "stabilityai/stable-diffusion-xl-refiner-1.0", "stabilityai/sdxl-turbo", diff --git a/benchmarks/benchmark_sd_inpainting.py b/benchmarks/benchmark_sd_inpainting.py index 8f36883e16f3..143adcb0d87c 100644 --- a/benchmarks/benchmark_sd_inpainting.py +++ b/benchmarks/benchmark_sd_inpainting.py @@ -11,9 +11,9 @@ parser.add_argument( "--ckpt", type=str, - default="runwayml/stable-diffusion-v1-5", + default="Lykon/DreamShaper", choices=[ - "runwayml/stable-diffusion-v1-5", + "Lykon/DreamShaper", "stabilityai/stable-diffusion-2-1", "stabilityai/stable-diffusion-xl-base-1.0", ], diff --git a/benchmarks/benchmark_text_to_image.py b/benchmarks/benchmark_text_to_image.py index caa97b0c5e3b..ddc7fb2676a5 100644 --- a/benchmarks/benchmark_text_to_image.py +++ b/benchmarks/benchmark_text_to_image.py @@ -7,7 +7,7 @@ ALL_T2I_CKPTS = [ - "runwayml/stable-diffusion-v1-5", + "Lykon/DreamShaper", "segmind/SSD-1B", "stabilityai/stable-diffusion-xl-base-1.0", "kandinsky-community/kandinsky-2-2-decoder", @@ -21,7 +21,7 @@ parser.add_argument( "--ckpt", type=str, - default="runwayml/stable-diffusion-v1-5", + default="Lykon/DreamShaper", choices=ALL_T2I_CKPTS, ) parser.add_argument("--batch_size", type=int, default=1)