Make caption naturalizer graph-friendly

This commit is contained in:
2026-06-25 13:57:08 +02:00
parent 6f294662be
commit 57587fccf3
+8 -4
View File
@@ -266,8 +266,9 @@ NODE_INPUT_TOOLTIPS = {
"nude_weight": "Weight used when explicit nude/body exposure tags are inferred.",
},
"SxCPCaptionNaturalizer": {
"style_policy": "drop_style_tail removes training/style boilerplate; keep_style_terms preserves more of it.",
"include_trigger": "Add the naturalizer trigger to the rewritten caption.",
"metadata_json": "Best input for training captions because it preserves structured generator details.",
"style_policy": "drop_style_tail removes generation/style boilerplate; keep_style_terms preserves more of it.",
"include_trigger": "Keep this true for LoRA/training captions so the trigger token is learned.",
},
"SxCPForLoopStart": {
"index": "Output loop index. First generated index is skip + 1.",
@@ -2106,7 +2107,8 @@ class SxCPCaptionNaturalizer:
"include_trigger": ("BOOLEAN", {"default": True}),
},
"optional": {
"metadata_json": ("STRING", {"default": "", "multiline": True}),
"source_text_input": ("STRING", {"forceInput": True}),
"metadata_json": ("STRING", {"forceInput": True}),
},
}
@@ -2123,10 +2125,12 @@ class SxCPCaptionNaturalizer:
style_policy,
trigger,
include_trigger,
source_text_input="",
metadata_json="",
):
active_source_text = source_text_input or source_text or ""
return naturalize_caption(
source_text=source_text or "",
source_text=active_source_text,
metadata_json=metadata_json or "",
input_hint=input_hint,
trigger=trigger,