filtering in settings paint/inpainting models in dropdown

pull/75/head
GeorgLegato 2023-04-28 03:22:54 +02:00
parent 44114c2252
commit f48d29b1be
1 changed files with 2 additions and 2 deletions

View File

@ -67,7 +67,7 @@ def on_ui_settings():
None, None,
"Name of your desired model to render keyframes (txt2img)", "Name of your desired model to render keyframes (txt2img)",
gr.Dropdown, 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, section=section,
), ),
) )
@ -78,7 +78,7 @@ def on_ui_settings():
None, None,
"Name of your desired inpaint model (img2img-inpaint). Default is vanilla sd-v1-5-inpainting.ckpt ", "Name of your desired inpaint model (img2img-inpaint). Default is vanilla sd-v1-5-inpainting.ckpt ",
gr.Dropdown, 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, section=section,
), ),
) )