From 4e28d803105b56f8c979f0eb28f49cc1fcf8c30d Mon Sep 17 00:00:00 2001 From: Dowon Date: Wed, 10 Jan 2024 22:43:25 +0900 Subject: [PATCH] fix: `inpaint_depth` controlnet model exception --- adetailer/ui.py | 1 + 1 file changed, 1 insertion(+) diff --git a/adetailer/ui.py b/adetailer/ui.py index 539ce82..57f789f 100644 --- a/adetailer/ui.py +++ b/adetailer/ui.py @@ -71,6 +71,7 @@ def on_generate_click(state: dict, *values: Any): def on_cn_model_update(cn_model_name: str): + cn_model_name = cn_model_name.replace("inpaint_depth", "depth") for t in cn_module_choices: if t in cn_model_name: choices = cn_module_choices[t]