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

Fix setting scaling factor in VAE config (#6779)

Browse Source 
fix
This commit is contained in:
Dhruv Nair
2024-01-31 19:47:22 +05:30
committed by GitHub
parent 97ee616971
commit d7dc0ffd79
2 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -61,8 +61,9 @@ def build_sub_model_components(
return unet_components
if component_name == "vae":
scaling_factor = kwargs.get("scaling_factor", None)
vae_components = create_diffusers_vae_model_from_ldm(
pipeline_class_name, original_config, checkpoint, image_size
pipeline_class_name, original_config, checkpoint, image_size, scaling_factor
)
return vae_components

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

@@ -513,11 +513,12 @@ def create_controlnet_diffusers_config(original_config, image_size: int):
return controlnet_config
def create_vae_diffusers_config(original_config, image_size, scaling_factor=0.18125):
def create_vae_diffusers_config(original_config, image_size, scaling_factor=None):
"""
Creates a config for the diffusers based on the config of the LDM model.
"""
vae_params = original_config["model"]["params"]["first_stage_config"]["params"]["ddconfig"]
scaling_factor = scaling_factor or original_config["model"]["params"]["scale_factor"]
block_out_channels = [vae_params["ch"] * mult for mult in vae_params["ch_mult"]]
down_block_types = ["DownEncoderBlock2D"] * len(block_out_channels)