fix: revert CNHijackRestore

pull/556/head
Dowon 2024-03-16 23:02:58 +09:00
parent c3e2884b80
commit 4ec348a462
1 changed files with 1 additions and 6 deletions

View File

@ -13,12 +13,7 @@ except ImportError:
get_cn_models,
)
from .restore import cn_allow_script_control
if controlnet_type == "forge":
from contextlib import nullcontext as CNHijackRestore # noqa: N812
else:
from .restore import CNHijackRestore
from .restore import CNHijackRestore, cn_allow_script_control
__all__ = [
"ControlNetExt",