From 57587fccf33c0f37c23d53cc98920bfeddc29280 Mon Sep 17 00:00:00 2001 From: Ethanfel Date: Thu, 25 Jun 2026 13:57:08 +0200 Subject: [PATCH] Make caption naturalizer graph-friendly --- __init__.py | 12 ++++++++---- 1 file changed, 8 insertions(+), 4 deletions(-) diff --git a/__init__.py b/__init__.py index e008a5e..3d38ec1 100644 --- a/__init__.py +++ b/__init__.py @@ -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,