From f94b453d10c2eb4e3ad57b7cdfdd17e60a557b3e Mon Sep 17 00:00:00 2001
From: Mikubill <31246794+Mikubill@users.noreply.github.com>
Date: Fri, 24 Feb 2023 03:22:41 +0000
Subject: [PATCH] fix: allow model config override (#331)

---
 scripts/controlnet.py | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/scripts/controlnet.py b/scripts/controlnet.py
index 310def4..de1c5ce 100644
--- a/scripts/controlnet.py
+++ b/scripts/controlnet.py
@@ -430,10 +430,15 @@ class Script(scripts.Script):
         state_dict = load_state_dict(model_path)
         network_module = PlugableControlModel
         network_config = shared.opts.data.get("control_net_model_config", default_conf)
+
         if any([k.startswith("body.") for k, v in state_dict.items()]):
             # adapter model     
             network_module = PlugableAdapter
             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(
             state_dict=state_dict,