Bing-su/adetailer
1
0
Fork 0
mirror of https://github.com/Bing-su/adetailer.git synced 2026-03-11 16:30:03 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: continue when NansException raised

Browse Source
This commit is contained in:
Bingsu
2023-07-02 20:11:03 +09:00
parent 01eaf329ce
commit f6b76e8fae
1 changed files with 16 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

30
scripts/!adetailer.py
View File

@@ -542,23 +542,25 @@ class AfterDetailerScript(scripts.Script):
p2.image_mask = masks[j]
self.i2i_prompts_replace(p2, ad_prompts, ad_negatives, j)
if not re.match(r"^\s*\[SKIP\]\s*$", p2.prompt):
if args.ad_controlnet_model == "None":
cn_restore_unet_hook(p2, self.cn_latest_network)
if re.match(r"^\s*\[SKIP\]\s*$", p2.prompt):
continue
try:
processed = process_images(p2)
except NansException as e:
msg = f"[-] ADetailer: 'NansException' occurred with {ordinal(n + 1)} settings.\n{e}"
print(msg, file=sys.stderr)
return False
p2.seed = seed + j
p2.subseed = subseed + j
self.compare_prompt(p2, processed, n=n)
p2 = copy(i2i)
p2.init_images = [processed.images[0]]
if args.ad_controlnet_model == "None":
cn_restore_unet_hook(p2, self.cn_latest_network)
p2.seed = seed + j + 1
p2.subseed = subseed + j + 1
try:
processed = process_images(p2)
except NansException as e:
msg = f"[-] ADetailer: 'NansException' occurred with {ordinal(n + 1)} settings.\n{e}"
print(msg, file=sys.stderr)
continue
self.compare_prompt(p2, processed, n=n)
p2 = copy(i2i)
p2.init_images = [processed.images[0]]
if processed is not None:
pp.image = processed.images[0]