nodes/MinusZoneAI-ComfyUI-Prompt-MZ
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

'update'

Browse Source
This commit is contained in:
wailovet
2024-06-26 01:07:57 +08:00
parent 06fc84c2ed
commit a1566f84dd
1 changed files with 25 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
__init__.py
+25 -2
View File
@@ -649,8 +649,11 @@ class MZ_Florence2Captioner:
"Florence-2-large-ft",
"Florence-2-large",
],),
"directory": ("STRING", {"default": "", "placeholder": "directory"}),
"resolution": ("INT", {"default": 512, "min": 128, "max": 0xffffffffffffffff}),
"captioner_config": ("ImageCaptionerConfig", ),
"caption_suffix": ("STRING", {"default": ".caption"}),
"force_update": ([False, True], {"default": False}),
"prompt_fixed_beginning": ("STRING", {"default": "", }),
},
"optional": {
},
@@ -666,6 +669,16 @@ class MZ_Florence2Captioner:
kwargs = kwargs.copy()
from . import mz_transformers
importlib.reload(mz_transformers)
kwargs["captioner_config"] = {
"directory": kwargs["directory"],
"resolution": kwargs["resolution"],
"caption_suffix": kwargs["caption_suffix"],
"force_update": kwargs["force_update"],
"prompt_fixed_beginning": kwargs["prompt_fixed_beginning"],
}
return mz_transformers.florence2_node_encode(kwargs)
@@ -721,8 +734,11 @@ class MZ_PaliGemmaCaptioner:
"paligemma-sd3-long-captioner-v2",
"paligemma-sd3-long-captioner",
],),
"directory": ("STRING", {"default": "", "placeholder": "directory"}),
"resolution": ("INT", {"default": 512, "min": 128, "max": 0xffffffffffffffff}),
"captioner_config": ("ImageCaptionerConfig", ),
"caption_suffix": ("STRING", {"default": ".caption"}),
"force_update": ([False, True], {"default": False}),
"prompt_fixed_beginning": ("STRING", {"default": "", }),
},
"optional": {
},
@@ -738,6 +754,13 @@ class MZ_PaliGemmaCaptioner:
kwargs = kwargs.copy()
from . import mz_transformers
importlib.reload(mz_transformers)
kwargs["captioner_config"] = {
"directory": kwargs["directory"],
"resolution": kwargs["resolution"],
"caption_suffix": kwargs["caption_suffix"],
"force_update": kwargs["force_update"],
"prompt_fixed_beginning": kwargs["prompt_fixed_beginning"],
}
return mz_transformers.paligemma_node_encode(kwargs)