diff options
| author | brkirch <brkirch@users.noreply.github.com> | 2023-01-04 23:10:31 -0500 |
|---|---|---|
| committer | brkirch <brkirch@users.noreply.github.com> | 2023-01-06 00:15:22 -0500 |
| commit | 3bfe2bb5498241c4873cdd71b3f0a5bac5f64d7f (patch) | |
| tree | c296183ec50286cf3552add8c1c967c2f843ec47 /modules/sd_hijack_inpainting.py | |
| parent | f6ab5a39d762a7791573d1c52ae5a3024b10e8ed (diff) | |
| parent | 5f4fa942b8ec3ed3b15a352903489d6f9e6eb46e (diff) | |
Merge remote-tracking branch 'upstream/master' into sub-quad_attn_opt
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 |