From 5fc736c910ffa301e5e5aa55625b5a1c9d9856fa Mon Sep 17 00:00:00 2001 From: Bingsu Date: Sat, 29 Jul 2023 00:47:23 +0900 Subject: [PATCH] fix: remove unused cn --- controlnet_ext/restore.py | 6 ------ scripts/!adetailer.py | 5 ----- 2 files changed, 11 deletions(-) diff --git a/controlnet_ext/restore.py b/controlnet_ext/restore.py index 5b9bfa6..152ffd0 100644 --- a/controlnet_ext/restore.py +++ b/controlnet_ext/restore.py @@ -5,12 +5,6 @@ from contextlib import contextmanager from modules import img2img, processing, shared -def cn_restore_unet_hook(p, cn_latest_network): - if cn_latest_network is not None: - unet = p.sd_model.model.diffusion_model - cn_latest_network.restore(unet) - - class CNHijackRestore: def __init__(self): self.process = hasattr(processing, "__controlnet_original_process_images_inner") diff --git a/scripts/!adetailer.py b/scripts/!adetailer.py index 493129a..a5b2be5 100644 --- a/scripts/!adetailer.py +++ b/scripts/!adetailer.py @@ -101,8 +101,6 @@ class AfterDetailerScript(scripts.Script): self.ultralytics_device = self.get_ultralytics_device() self.controlnet_ext = None - self.cn_script = None - self.cn_latest_network = None def __repr__(self): return f"{self.__class__.__name__}(version={__version__})" @@ -347,9 +345,6 @@ class AfterDetailerScript(scripts.Script): filename = Path(filepath).stem 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 return script_runner, script_args