Merge pull request #652 from DiligentCatCat/main

feat: use the closest model if model is not found
AUTOMATIC1111 · Mar 23, 2023 · 241c05f · 241c05f
2 parents 2f0e673 + b8fdfb1
commit 241c05f
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/scripts/controlnet.py b/scripts/controlnet.py
@@ -475,6 +475,10 @@ def load_control_model(self, p, unet, model, lowvram):
     def build_control_model(self, p, unet, model, lowvram):
 
         model_path = global_state.cn_models.get(model, None)
+        if model_path is None:
+            model = find_closest_lora_model_name(model)
+            model_path = global_state.cn_models.get(model, None)
+
         if model_path is None:
             raise RuntimeError(f"model not found: {model}")