diff --git a/modules/config.py b/modules/config.py index eef986ac..11481bb9 100644 --- a/modules/config.py +++ b/modules/config.py @@ -387,7 +387,7 @@ default_black_out_nsfw = get_config_item_or_set_default( 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))] diff --git a/modules/flags.py b/modules/flags.py index 83c67753..ad2acfee 100644 --- a/modules/flags.py +++ b/modules/flags.py @@ -48,7 +48,7 @@ performance_selections = [ output_formats = ['png', 'jpg', 'webp'] 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)'