Add options to u2net_cloth_seg

This commit is contained in:
rayronvictor
2024-01-24 21:26:23 -03:00
parent dea4b6c15c
commit f0fb9783e1
4 changed files with 51 additions and 23 deletions
+6
View File
@@ -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 = [
+3 -1
View File
@@ -36,9 +36,11 @@ inpaint_engine_versions = ['None', 'v1', 'v2.5', 'v2.6']
performance_selections = ['Speed', 'Quality', 'Extreme Speed']
inpaint_mask_models = [
'u2net', 'u2netp', 'silueta', 'isnet-general-use', 'isnet-anime'
'u2net', 'u2netp', 'u2net_human_seg', 'u2net_cloth_seg', 'silueta', 'isnet-general-use', 'isnet-anime'
]
inpaint_mask_cloth_category = ['full', 'upper', 'lower']
inpaint_option_default = 'Inpaint or Outpaint (default)'
inpaint_option_detail = 'Improve Detail (face, hand, eyes, etc.)'
inpaint_option_modify = 'Modify Content (add objects, change background, etc.)'