Bing-su/adetailer
1
0
Fork 0
mirror of https://github.com/Bing-su/adetailer.git synced 2026-01-26 19:29:54 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat: re-process controlnet

Browse Source
This commit is contained in:
Bingsu
2023-05-24 23:54:54 +09:00
parent f24137a570
commit 2f6b3c698b
1 changed files with 7 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
scripts/!adetailer.py
View File

@@ -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)