Mirrors
/
stable-diffusion-webui
mirror of https://github.com/AUTOMATIC1111/stable-diffusion-webui.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

nit

This commit is contained in:
huchenlei 2024-06-08 22:15:37 -04:00
parent d875cda565
commit f89b5dbbd2
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
modules/sd_models.py
View File

@ -402,8 +402,9 @@ def load_model_weights(model, checkpoint_info: CheckpointInfo, state_dict, timer
del state_dict del state_dict
# Set is_sdxl_inpaint flag. # Set is_sdxl_inpaint flag.
# Perform this check after model initialization to make sure state_dict # Checks Unet structure to detect inpaint model. The inpaint model's
# structure is already known. # checkpoint state_dict does not contain the key
# 'diffusion_model.input_blocks.0.0.weight'.
diffusion_model_input = model.model.state_dict().get( diffusion_model_input = model.model.state_dict().get(
'diffusion_model.input_blocks.0.0.weight' 'diffusion_model.input_blocks.0.0.weight'
) )