1580158
/
sd-webui-controlnet
mirror of https://github.com/Mikubill/sd-webui-controlnet.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

fix: allow model config override (#331)

pull/340/head
Mikubill 2023-02-24 03:22:41 +00:00
parent ecad77ed65
commit f94b453d10
1 changed files with 5 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
scripts/controlnet.py
View File

@ -430,10 +430,15 @@ class Script(scripts.Script):
state_dict = load_state_dict(model_path) state_dict = load_state_dict(model_path)
network_module = PlugableControlModel network_module = PlugableControlModel
network_config = shared.opts.data.get("control_net_model_config", default_conf) network_config = shared.opts.data.get("control_net_model_config", default_conf)
if any([k.startswith("body.") for k, v in state_dict.items()]): if any([k.startswith("body.") for k, v in state_dict.items()]):
# adapter model # adapter model
network_module = PlugableAdapter network_module = PlugableAdapter
network_config = shared.opts.data.get("control_net_model_adapter_config", default_conf_adapter) network_config = shared.opts.data.get("control_net_model_adapter_config", default_conf_adapter)
override_config = os.path.splitext(model_path)[0] + ".yaml"
if os.path.exists(override_config):
network_config = override_config
network = network_module( network = network_module(
state_dict=state_dict, state_dict=state_dict,