Skip to content

S
Stable Diffusion Webui
Commit f89b5dbb authored by huchenlei's avatar huchenlei
Browse files
Options

nit

parent d875cda5
Show whitespace changes
Inline Side-by-side
Showing with 3 additions and 2 deletions
modules/sd_models.py
View file @ f89b5dbb
...@@ -402,8 +402,9 @@ def load_model_weights(model, checkpoint_info: CheckpointInfo, state_dict, timer ...@@ -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'
) )
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment