diff options
| author | AUTOMATIC1111 <16777216c@gmail.com> | 2022-12-24 12:19:06 +0300 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2022-12-24 12:19:06 +0300 |
| commit | 8c9e6d3c7d3e4f592d9154ec45533ef47d71ef88 (patch) | |
| tree | 79abeec21ca12b30a6aada833757fb9e790eec60 /modules | |
| parent | 67ff058b8d702569c8b096234b55f75e767ea93a (diff) | |
| parent | 9146a5884cbdf67c019685950f7ad0b3f7bd9230 (diff) | |
Merge pull request #5131 from uservar/inpainting-detection
Better should_hijack_inpainting detection
Diffstat (limited to 'modules')
| -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 85e7281f..bb5499b3 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(): |