fix: replace CNHijackRestore with nullcontext for forge

controlnet_ext/__init__.py

@@ -13,8 +13,17 @@ 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
+
 __all__ = [
     "ControlNetExt",
+    "CNHijackRestore",
+    "cn_allow_script_control",
     "controlnet_exists",
     "controlnet_type",
     "get_cn_models",