From ce277b89d8ba8592205cba319fb7ac49abbc95df Mon Sep 17 00:00:00 2001 From: GeorgLegato Date: Fri, 28 Apr 2023 03:22:54 +0200 Subject: [PATCH] filtering in settings paint/inpainting models in dropdown --- iz_helpers/settings.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/iz_helpers/settings.py b/iz_helpers/settings.py index 4338526..e201588 100644 --- a/iz_helpers/settings.py +++ b/iz_helpers/settings.py @@ -67,7 +67,7 @@ def on_ui_settings(): None, "Name of your desired model to render keyframes (txt2img)", gr.Dropdown, - lambda: {"choices": shared.list_checkpoint_tiles()}, + lambda: {"choices": [x for x in list(shared.list_checkpoint_tiles()) if "inpainting" not in x]}, section=section, ), ) @@ -78,7 +78,7 @@ def on_ui_settings(): None, "Name of your desired inpaint model (img2img-inpaint). Default is vanilla sd-v1-5-inpainting.ckpt ", gr.Dropdown, - lambda: {"choices": shared.list_checkpoint_tiles()}, + lambda: {"choices": [x for x in list(shared.list_checkpoint_tiles()) if "inpainting" in x]}, section=section, ), )