fix: revert CNHijackRestore

This commit is contained in:
Dowon
2024-03-16 23:02:58 +09:00
parent c3e2884b80
commit 4ec348a462

View File

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