From 5daf61a34288cc24b5936434df3be6772084071c Mon Sep 17 00:00:00 2001 From: Dhruv Nair Date: Sat, 30 Dec 2023 08:24:26 +0000 Subject: [PATCH] update --- src/diffusers/loaders/single_file_utils.py | 2 +- src/diffusers/pipelines/stable_diffusion/convert_from_ckpt.py | 1 + 2 files changed, 2 insertions(+), 1 deletion(-) diff --git a/src/diffusers/loaders/single_file_utils.py b/src/diffusers/loaders/single_file_utils.py index 5598c786ba..af3b2b7fa8 100644 --- a/src/diffusers/loaders/single_file_utils.py +++ b/src/diffusers/loaders/single_file_utils.py @@ -224,7 +224,7 @@ def determine_image_size(pipeline_class_name, original_config, checkpoint, **kwa model_type = infer_model_type(pipeline_class_name, original_config, **kwargs) if pipeline_class_name == "StableDiffusionUpscalePipeline": - image_size = image_size or original_config.model.params.unet_config.params.image_size + image_size = original_config.model.params.unet_config.params.image_size if model_type in ["SDXL", "SDXL-Refiner"]: image_size = 1024 diff --git a/src/diffusers/pipelines/stable_diffusion/convert_from_ckpt.py b/src/diffusers/pipelines/stable_diffusion/convert_from_ckpt.py index 5aa23252b8..d80f4dc863 100644 --- a/src/diffusers/pipelines/stable_diffusion/convert_from_ckpt.py +++ b/src/diffusers/pipelines/stable_diffusion/convert_from_ckpt.py @@ -1436,6 +1436,7 @@ def download_from_original_stable_diffusion_ckpt( if pipeline_class == StableDiffusionUpscalePipeline: image_size = original_config.model.params.unet_config.params.image_size + import ipdb; ipdb.set_trace() # Convert the UNet2DConditionModel model. unet_config = create_unet_diffusers_config(original_config, image_size=image_size)