fix: correctly validate default_inpaint_mask_sam_model

This commit is contained in:
Manuel Schmid
2024-08-08 13:15:15 +02:00
parent ee12d114c1
commit 8bd9ea1dbf
+1 -1
View File
@@ -702,7 +702,7 @@ default_inpaint_mask_cloth_category = get_config_item_or_set_default(
default_inpaint_mask_sam_model = get_config_item_or_set_default(
key='default_inpaint_mask_sam_model',
default_value='vit_b',
validator=lambda x: x in [y[1] for y in modules.flags.inpaint_mask_sam_model if y[1] == x],
validator=lambda x: x in modules.flags.inpaint_mask_sam_model,
expected_type=str
)