diff options
| author | MrCheeze <fishycheeze@yahoo.ca> | 2022-12-10 16:29:26 +0000 |
|---|---|---|
| committer | MrCheeze <fishycheeze@yahoo.ca> | 2022-12-10 16:29:26 +0000 |
| commit | bd81a09eacf02dad095b98094ab936f276d0343f (patch) | |
| tree | e7fe2566b861c67363bf971b6cc3ff29723d99be /modules/sd_models.py | |
| parent | a1c8ad88283f7b3e861e4722c71e39bf71eec744 (diff) | |
| download | stable-diffusion-webui-gfx803-bd81a09eacf02dad095b98094ab936f276d0343f.tar.gz stable-diffusion-webui-gfx803-bd81a09eacf02dad095b98094ab936f276d0343f.tar.bz2 stable-diffusion-webui-gfx803-bd81a09eacf02dad095b98094ab936f276d0343f.zip | |
fix support for 2.0 inpainting model while maintaining support for 1.5 inpainting model
Diffstat (limited to 'modules/sd_models.py')
| -rw-r--r-- | modules/sd_models.py | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/modules/sd_models.py b/modules/sd_models.py index 5b37f3fe..b64f573f 100644 --- a/modules/sd_models.py +++ b/modules/sd_models.py @@ -296,6 +296,7 @@ def load_model(checkpoint_info=None): sd_config.model.params.use_ema = False
sd_config.model.params.conditioning_key = "hybrid"
sd_config.model.params.unet_config.params.in_channels = 9
+ sd_config.model.params.finetune_keys = None
# Create a "fake" config with a different name so that we know to unload it when switching models.
checkpoint_info = checkpoint_info._replace(config=checkpoint_info.config.replace(".yaml", "-inpainting.yaml"))
|