Forge webui support (#517)

* Forge support * Separate Forge implementation * Remove debug print
2026-04-29 10:41:14 +00:00 · 2024-02-27 07:19:36 -05:00
parent ac0bb9ab00
commit 1441b55d6f
6 changed files with 171 additions and 37 deletions
--- a/controlnet_ext/init.py
+++ b/controlnet_ext/init.py
@@ -1,7 +1,21 @@
-from .controlnet_ext import ControlNetExt, controlnet_exists, get_cn_models
+try:
+    from .controlnet_ext_forge import (
+        ControlNetExt,
+        controlnet_exists,
+        controlnet_forge,
+        get_cn_models,
+    )
+except ImportError:
+    from .controlnet_ext import (
+        ControlNetExt,
+        controlnet_exists,
+        controlnet_forge,
+        get_cn_models,
+    )

 __all__ = [
    "ControlNetExt",
    "controlnet_exists",
+    "controlnet_forge",
    "get_cn_models",
 ]