feat: add handling for stage2_mask_sam_max_num_boxes and config

This commit is contained in:
Manuel Schmid
2024-06-12 22:16:02 +02:00
parent 9998b52dd2
commit dbc844804b
3 changed files with 33 additions and 18 deletions
+8 -2
View File
@@ -510,12 +510,18 @@ example_stage2_prompts = get_config_item_or_set_default(
validator=lambda x: isinstance(x, list) and all(isinstance(v, str) for v in x),
expected_type=list
)
default_max_stage2_tabs = get_config_item_or_set_default(
key='default_max_stage2_tabs',
default_stage2_tabs = get_config_item_or_set_default(
key='default_stage2_tabs',
default_value=3,
validator=lambda x: isinstance(x, int) and 1 <= x <= 5,
expected_type=int
)
default_sam_max_num_boxes = get_config_item_or_set_default(
key='default_sam_max_num_boxes',
default_value=2,
validator=lambda x: isinstance(x, int) and 1 <= x <= 5,
expected_type=int
)
default_black_out_nsfw = get_config_item_or_set_default(
key='default_black_out_nsfw',
default_value=False,