mirror of
https://github.com/Bing-su/adetailer.git
synced 2026-03-13 01:10:01 +00:00
feat: ignore NansException
This commit is contained in:
@@ -38,6 +38,7 @@ from controlnet_ext.restore import (
|
||||
cn_restore_unet_hook,
|
||||
)
|
||||
from sd_webui import images, safe, script_callbacks, scripts, shared
|
||||
from sd_webui.devices import NansException
|
||||
from sd_webui.paths import data_path, models_path
|
||||
from sd_webui.processing import (
|
||||
StableDiffusionProcessingImg2Img,
|
||||
@@ -541,7 +542,13 @@ class AfterDetailerScript(scripts.Script):
|
||||
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)
|
||||
processed = process_images(p2)
|
||||
|
||||
try:
|
||||
processed = process_images(p2)
|
||||
except NansException as e:
|
||||
msg = f"[-] ADetailer: 'NansException' occurred with {ordinal(n + 1)} settings.\n{e}"
|
||||
print(msg)
|
||||
return False
|
||||
|
||||
self.compare_prompt(p2, processed)
|
||||
p2 = copy(i2i)
|
||||
|
||||
11
sd_webui/devices.py
Normal file
11
sd_webui/devices.py
Normal file
@@ -0,0 +1,11 @@
|
||||
from __future__ import annotations
|
||||
|
||||
from typing import TYPE_CHECKING
|
||||
|
||||
if TYPE_CHECKING:
|
||||
|
||||
class NansException(Exception): # noqa: N818
|
||||
pass
|
||||
|
||||
else:
|
||||
from modules.devices import NansException
|
||||
Reference in New Issue
Block a user