diff options
| author | uservar <63248296+uservar@users.noreply.github.com> | 2022-11-27 19:11:50 +0000 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2022-11-27 19:11:50 +0000 |
| commit | 9146a5884cbdf67c019685950f7ad0b3f7bd9230 (patch) | |
| tree | bbc70f8d70eb561bdc19061c19fe285a53e8359d /modules/sd_hijack_inpainting.py | |
| parent | 9e78d2c419732711e984c4478af15ece121d64fd (diff) | |
| download | stable-diffusion-webui-gfx803-9146a5884cbdf67c019685950f7ad0b3f7bd9230.tar.gz stable-diffusion-webui-gfx803-9146a5884cbdf67c019685950f7ad0b3f7bd9230.tar.bz2 stable-diffusion-webui-gfx803-9146a5884cbdf67c019685950f7ad0b3f7bd9230.zip | |
Better should hijack inpainting detection
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 938f9a58..5dcbbed9 100644 --- a/modules/sd_hijack_inpainting.py +++ b/modules/sd_hijack_inpainting.py @@ -1,3 +1,4 @@ +import os import torch from einops import repeat @@ -319,7 +320,9 @@ class LatentInpaintDiffusion(LatentDiffusion): def should_hijack_inpainting(checkpoint_info): - return str(checkpoint_info.filename).endswith("inpainting.ckpt") and not checkpoint_info.config.endswith("inpainting.yaml") + ckpt_basename = os.path.basename(checkpoint_info.filename).lower() + cfg_basename = os.path.basename(checkpoint_info.config).lower() + return "inpainting" in ckpt_basename and not "inpainting" in cfg_basename def do_inpainting_hijack(): |