AI/diffusers
1
0
Fork 0
mirror of https://github.com/huggingface/diffusers.git synced 2026-01-29 07:22:12 +03:00
Code Activity

update

Browse Source
This commit is contained in:
Dhruv Nair
2023-12-30 08:24:26 +00:00
parent 658d80f3cf
commit 5daf61a342
2 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/diffusers/loaders/single_file_utils.py
View File

@@ -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

1
src/diffusers/pipelines/stable_diffusion/convert_from_ckpt.py
View File

@@ -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)