From 09221b1065d4ed889c0fab485f6358909acf404b Mon Sep 17 00:00:00 2001 From: Dowon Date: Sat, 30 Dec 2023 16:30:20 +0900 Subject: [PATCH] feat: auto disable adetailer when inpainting --- scripts/!adetailer.py | 10 ++++++++++ 1 file changed, 10 insertions(+) diff --git a/scripts/!adetailer.py b/scripts/!adetailer.py index 9600e27..f23845c 100644 --- a/scripts/!adetailer.py +++ b/scripts/!adetailer.py @@ -656,11 +656,21 @@ class AfterDetailerScript(scripts.Script): use_same_seed = shared.opts.data.get("ad_same_seed_for_each_tap", False) return seed if use_same_seed else seed + i + @staticmethod + def is_img2img_inpaint(p) -> bool: + return hasattr(p, "image_mask") and bool(p.image_mask) + @rich_traceback def process(self, p, *args_): if getattr(p, "_ad_disabled", False): return + if self.is_img2img_inpaint(p): + p._ad_disabled = True + msg = "[-] ADetailer: img2img inpainting detected. adetailer disabled." + print(msg) + return + if self.is_ad_enabled(*args_): arg_list = self.get_args(p, *args_) self.check_skip_img2img(p, *args_)