From 2f6b3c698b65c9ca251bf62563b2faedccaa26ca Mon Sep 17 00:00:00 2001 From: Bingsu Date: Wed, 24 May 2023 23:54:54 +0900 Subject: [PATCH] feat: re-process controlnet --- scripts/!adetailer.py | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/scripts/!adetailer.py b/scripts/!adetailer.py index 4d2e574..d5dda67 100644 --- a/scripts/!adetailer.py +++ b/scripts/!adetailer.py @@ -74,8 +74,10 @@ class ChangeTorchLoad: class AfterDetailerScript(scripts.Script): def __init__(self): super().__init__() - self.controlnet_ext = None self.ultralytics_device = self.get_ultralytics_device() + + self.controlnet_ext = None + self.cn_script = None self.cn_latest_network = None def title(self): @@ -295,6 +297,7 @@ class AfterDetailerScript(scripts.Script): if filename in script_names_set: filtered_alwayson.append(script_object) if filename == "controlnet": + self.cn_script = script_object self.cn_latest_network = script_object.latest_network script_runner.alwayson_scripts = filtered_alwayson @@ -515,6 +518,9 @@ class AfterDetailerScript(scripts.Script): p, init_image, condition="ad_save_images_before", suffix="-ad-before" ) + if self.cn_script is not None: + self.cn_script.process(p) + try: if p._idx == len(p.all_prompts) - 1: self.write_params_txt(p)