fix: add correct flag validation for default_inpaint_mask_model config

This commit is contained in:
Manuel Schmid 2024-01-24 19:40:52 +01:00
parent d7d34647d0
commit dea4b6c15c
No known key found for this signature in database
GPG Key ID: 32C4F7569B40B84B
1 changed files with 1 additions and 1 deletions

View File

@ -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))]