diff --git a/nodes.py b/nodes.py index 8c53789..7b3d354 100644 --- a/nodes.py +++ b/nodes.py @@ -52,7 +52,9 @@ class DownloadAndLoadFlorence2Model: 'gokaygokay/Florence-2-SD3-Captioner', 'gokaygokay/Florence-2-Flux-Large', 'MiaoshouAI/Florence-2-base-PromptGen-v1.5', - 'MiaoshouAI/Florence-2-large-PromptGen-v1.5' + 'MiaoshouAI/Florence-2-large-PromptGen-v1.5', + 'MiaoshouAI/Florence-2-base-PromptGen-v2.0', + 'MiaoshouAI/Florence-2-large-PromptGen-v2.0' ], { "default": 'microsoft/Florence-2-base' @@ -208,7 +210,9 @@ class Florence2Run: 'ocr_with_region', 'docvqa', 'prompt_gen_tags', - 'prompt_gen_mixed_caption' + 'prompt_gen_mixed_caption', + 'prompt_gen_analyze', + 'prompt_gen_mixed_caption_plus', ], ), "fill_mask": ("BOOLEAN", {"default": True}), @@ -270,7 +274,9 @@ class Florence2Run: 'ocr_with_region': '', 'docvqa': '', 'prompt_gen_tags': '', - 'prompt_gen_mixed_caption': '' + 'prompt_gen_mixed_caption': '', + 'prompt_gen_analyze': '', + 'prompt_gen_mixed_caption_plus': '', } task_prompt = prompts.get(task, '')