Add options to u2net_cloth_seg

2026-08-16 13:13:16 +02:00 · 2024-01-24 21:26:23 -03:00
parent dea4b6c15c
commit f0fb9783e1
4 changed files with 51 additions and 23 deletions
@@ -324,6 +324,12 @@ default_inpaint_mask_model = get_config_item_or_set_default(
    validator=lambda x: x in modules.flags.inpaint_mask_models
 )

+default_inpaint_mask_cloth_category = get_config_item_or_set_default(
+    key='default_inpaint_mask_cloth_category',
+    default_value='full',
+    validator=lambda x: x in modules.flags.inpaint_mask_cloth_category
+)
+
 config_dict["default_loras"] = default_loras = default_loras[:5] + [['None', 1.0] for _ in range(5 - len(default_loras))]

 possible_preset_keys = [