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

fix: call cn process only needed

Browse Source
This commit is contained in:
Bingsu
2023-05-26 12:17:20 +09:00
parent 8ddb7dd7b7
commit abf2dd5401
1 changed files with 8 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
scripts/!adetailer.py
View File

@@ -434,6 +434,12 @@ class AfterDetailerScript(scripts.Script):
i2i.prompt = prompt
i2i.negative_prompt = negative_prompt
def is_need_call_process(self, p):
i = p._idx
n_iter = p.iteration
bs = p.batch_size
return (i == (n_iter + 1) * bs - 1) and (i != len(p.all_prompts) - 1)
def process(self, p, *args_):
if getattr(p, "_disable_adetailer", False):
return
@@ -541,8 +547,8 @@ 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)
if self.cn_script is not None and self.is_need_call_process(p):
self.cn_script.process(p)
try:
if p._idx == len(p.all_prompts) - 1: