happyz
/
Fooocus
mirror of https://github.com/lllyasviel/Fooocus.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Merge branch 'feature/add-inpaint-mask-generation'

This commit is contained in:
Manuel Schmid 2024-01-24 19:45:58 +01:00
parent e196a35ce2 dea4b6c15c
commit 23bb69bb13
No known key found for this signature in database
GPG Key ID: 32C4F7569B40B84B
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
modules/config.py
View File

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

2
modules/flags.py
View File

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