From b7732a07efd8fbb260c3bcd3a93b2a34e4dbc2d2 Mon Sep 17 00:00:00 2001 From: Dhruv Nair Date: Sat, 30 Dec 2023 09:21:12 +0000 Subject: [PATCH] update --- src/diffusers/loaders/single_file_utils.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/src/diffusers/loaders/single_file_utils.py b/src/diffusers/loaders/single_file_utils.py index d9c8b12d53..9ee22ad43f 100644 --- a/src/diffusers/loaders/single_file_utils.py +++ b/src/diffusers/loaders/single_file_utils.py @@ -222,9 +222,11 @@ def determine_image_size(pipeline_class_name, original_config, checkpoint, **kwa if pipeline_class_name == "StableDiffusionUpscalePipeline": image_size = original_config.model.params.unet_config.params.image_size + return image_size elif model_type in ["SDXL", "SDXL-Refiner"]: image_size = 1024 + return image_size elif ( "parameterization" in original_config["model"]["params"] @@ -233,6 +235,7 @@ def determine_image_size(pipeline_class_name, original_config, checkpoint, **kwa # NOTE: For stable diffusion 2 base one has to pass `image_size==512` # as it relies on a brittle global step parameter here image_size = 512 if global_step == 875000 else 768 + return image_size return image_size