From d7d34647d0091748960a1e0587db55ea15ec5d3c Mon Sep 17 00:00:00 2001 From: Manuel Schmid Date: Wed, 24 Jan 2024 19:38:12 +0100 Subject: [PATCH 1/2] fix: remove segmentation models from rembg model selection --- modules/flags.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/modules/flags.py b/modules/flags.py index ffdce78d..9af5bb1f 100644 --- a/modules/flags.py +++ b/modules/flags.py @@ -36,7 +36,7 @@ inpaint_engine_versions = ['None', 'v1', 'v2.5', 'v2.6'] performance_selections = ['Speed', 'Quality', 'Extreme Speed'] inpaint_mask_models = [ - 'u2net', 'u2netp', 'u2net_human_seg', 'u2net_cloth_seg', 'silueta', 'isnet-general-use', 'isnet-anime', 'sam' + 'u2net', 'u2netp', 'silueta', 'isnet-general-use', 'isnet-anime' ] inpaint_option_default = 'Inpaint or Outpaint (default)' From dea4b6c15c89e1d4071829fba228a5a767d08d83 Mon Sep 17 00:00:00 2001 From: Manuel Schmid Date: Wed, 24 Jan 2024 19:40:52 +0100 Subject: [PATCH 2/2] fix: add correct flag validation for default_inpaint_mask_model config --- modules/config.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/modules/config.py b/modules/config.py index 4149f151..f8caaef8 100644 --- a/modules/config.py +++ b/modules/config.py @@ -321,7 +321,7 @@ example_inpaint_prompts = [[x] for x in example_inpaint_prompts] default_inpaint_mask_model = get_config_item_or_set_default( key='default_inpaint_mask_model', default_value='isnet-general-use', - validator=lambda x: isinstance(x, str) + validator=lambda x: x in modules.flags.inpaint_mask_models ) config_dict["default_loras"] = default_loras = default_loras[:5] + [['None', 1.0] for _ in range(5 - len(default_loras))]