From cd0f8be86eb1fbb727a11250ccb5ddc40424e85a Mon Sep 17 00:00:00 2001 From: Dowon Date: Thu, 16 Nov 2023 00:39:08 +0900 Subject: [PATCH] fix: add controlnet path to sys.path --- controlnet_ext/controlnet_ext.py | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/controlnet_ext/controlnet_ext.py b/controlnet_ext/controlnet_ext.py index 524106f..37c5a71 100644 --- a/controlnet_ext/controlnet_ext.py +++ b/controlnet_ext/controlnet_ext.py @@ -2,6 +2,7 @@ from __future__ import annotations import importlib import re +import sys from functools import lru_cache from pathlib import Path from textwrap import dedent @@ -34,6 +35,13 @@ for extension in extensions.active(): cn_base_path = ".".join(controlnet_path.parts[-2:]) break +if controlnet_path is not None: + sd_webui_controlnet_path = controlnet_path.resolve().parent + if sd_webui_controlnet_path.stem in ("extensions", "extensions-builtin"): + target_path = str(sd_webui_controlnet_path.parent) + if target_path not in sys.path: + sys.path.append(target_path) + cn_model_module = { "inpaint": "inpaint_global_harmonious", "scribble": "t2ia_sketch_pidi",