Bing-su/adetailer
1
0
Fork 0
mirror of https://github.com/Bing-su/adetailer.git synced 2026-01-26 11:19:53 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: trace controlnet import error

Browse Source
This commit is contained in:
Bingsu
2023-05-05 16:07:05 +09:00
parent 298a43000e
commit 88c8a899b0
4 changed files with 18 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
controlnet_ext/controlnet_ext.py
View File

@@ -34,20 +34,16 @@ class ControlNetExt:
self.cn_available = False
self.external_cn = None
def init_controlnet(self) -> bool:
def init_controlnet(self):
if is_in_builtin:
import_path = "extensions-builtin.sd-webui-controlnet.scripts.external_code"
else:
import_path = "extensions.sd-webui-controlnet.scripts.external_code"
try:
self.external_cn = importlib.import_module(import_path, "external_code")
self.cn_available = True
models = self.external_cn.get_models()
self.cn_models.extend(m for m in models if "inpaint" in m)
return True
except ImportError:
return False
self.external_cn = importlib.import_module(import_path, "external_code")
self.cn_available = True
models = self.external_cn.get_models()
self.cn_models.extend(m for m in models if "inpaint" in m)
def _update_scripts_args(self, p, model: str, weight: float):
cn_units = [