From 4ec348a462c88e6b461f168d475dafb11dd2b30f Mon Sep 17 00:00:00 2001
From: Dowon <ks2515@naver.com>
Date: Sat, 16 Mar 2024 23:02:58 +0900
Subject: [PATCH] fix: revert CNHijackRestore

---
 controlnet_ext/__init__.py | 7 +------
 1 file changed, 1 insertion(+), 6 deletions(-)

diff --git a/controlnet_ext/__init__.py b/controlnet_ext/__init__.py
index 1f03bed..6da28d4 100644
--- a/controlnet_ext/__init__.py
+++ b/controlnet_ext/__init__.py
@@ -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",