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 09:21:12 +00:00
parent 6d743eff0b
commit b7732a07ef
1 changed files with 3 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

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