diff options
| author | AUTOMATIC <16777216c@gmail.com> | 2023-01-04 09:47:42 +0000 |
|---|---|---|
| committer | AUTOMATIC <16777216c@gmail.com> | 2023-01-04 09:47:42 +0000 |
| commit | 8d8a05a3bbb50fdfeab51679a919d2487bd97976 (patch) | |
| tree | da2ff1dfa2aeece787f45f38bf0003ba30b64b7d /modules/sd_hijack_inpainting.py | |
| parent | 02d7abf5141431b9a3a8a189bb3136c71abd5e79 (diff) | |
| download | stable-diffusion-webui-gfx803-8d8a05a3bbb50fdfeab51679a919d2487bd97976.tar.gz stable-diffusion-webui-gfx803-8d8a05a3bbb50fdfeab51679a919d2487bd97976.tar.bz2 stable-diffusion-webui-gfx803-8d8a05a3bbb50fdfeab51679a919d2487bd97976.zip | |
find configs for models at runtime rather than when starting
Diffstat (limited to 'modules/sd_hijack_inpainting.py')
| -rw-r--r-- | modules/sd_hijack_inpainting.py | 5 |
1 files changed, 4 insertions, 1 deletions
diff --git a/modules/sd_hijack_inpainting.py b/modules/sd_hijack_inpainting.py index 3c214a35..31d2c898 100644 --- a/modules/sd_hijack_inpainting.py +++ b/modules/sd_hijack_inpainting.py @@ -97,8 +97,11 @@ def p_sample_plms(self, x, c, t, index, repeat_noise=False, use_original_steps=F def should_hijack_inpainting(checkpoint_info): + from modules import sd_models + ckpt_basename = os.path.basename(checkpoint_info.filename).lower() - cfg_basename = os.path.basename(checkpoint_info.config).lower() + cfg_basename = os.path.basename(sd_models.find_checkpoint_config(checkpoint_info)).lower() + return "inpainting" in ckpt_basename and not "inpainting" in cfg_basename |