Add SAM support

This commit is contained in:
rayronvictor
2024-01-25 16:45:02 -03:00
parent f0fb9783e1
commit 228a0aaeea
6 changed files with 113 additions and 8 deletions
+6
View File
@@ -330,6 +330,12 @@ default_inpaint_mask_cloth_category = get_config_item_or_set_default(
validator=lambda x: x in modules.flags.inpaint_mask_cloth_category
)
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
)
config_dict["default_loras"] = default_loras = default_loras[:5] + [['None', 1.0] for _ in range(5 - len(default_loras))]
possible_preset_keys = [