Foreign-Mirrors/Fooocus
1
0
Fork 0
mirror of https://github.com/lllyasviel/Fooocus.git synced 2026-08-16 13:13:16 +02:00
Code Issues Packages Projects Releases Wiki Activity

feat: change default_inpaint_mask_sam_model to match sam model registry

Browse Source
This commit is contained in:
Manuel Schmid
2024-06-10 20:45:56 +02:00
parent ce1fb74270
commit 3873892b0a
3 changed files with 5 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/config.py
+3 -3
View File
@@ -551,8 +551,8 @@ 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='sam_vit_b_01ec64',
validator=lambda x: x in modules.flags.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],
expected_type=str
)
@@ -792,7 +792,7 @@ def downloading_safety_checker_model():
def download_sam_model(sam_model: str) -> str:
match sam_model:
case 'default', 'vit_b':
case 'vit_b':
return downloading_sam_vit_b()
case 'vit_l':
return downloading_sam_vit_l()