Make caption naturalizer graph-friendly
This commit is contained in:
+8
-4
@@ -266,8 +266,9 @@ NODE_INPUT_TOOLTIPS = {
|
|||||||
"nude_weight": "Weight used when explicit nude/body exposure tags are inferred.",
|
"nude_weight": "Weight used when explicit nude/body exposure tags are inferred.",
|
||||||
},
|
},
|
||||||
"SxCPCaptionNaturalizer": {
|
"SxCPCaptionNaturalizer": {
|
||||||
"style_policy": "drop_style_tail removes training/style boilerplate; keep_style_terms preserves more of it.",
|
"metadata_json": "Best input for training captions because it preserves structured generator details.",
|
||||||
"include_trigger": "Add the naturalizer trigger to the rewritten caption.",
|
"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": {
|
"SxCPForLoopStart": {
|
||||||
"index": "Output loop index. First generated index is skip + 1.",
|
"index": "Output loop index. First generated index is skip + 1.",
|
||||||
@@ -2106,7 +2107,8 @@ class SxCPCaptionNaturalizer:
|
|||||||
"include_trigger": ("BOOLEAN", {"default": True}),
|
"include_trigger": ("BOOLEAN", {"default": True}),
|
||||||
},
|
},
|
||||||
"optional": {
|
"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,
|
style_policy,
|
||||||
trigger,
|
trigger,
|
||||||
include_trigger,
|
include_trigger,
|
||||||
|
source_text_input="",
|
||||||
metadata_json="",
|
metadata_json="",
|
||||||
):
|
):
|
||||||
|
active_source_text = source_text_input or source_text or ""
|
||||||
return naturalize_caption(
|
return naturalize_caption(
|
||||||
source_text=source_text or "",
|
source_text=active_source_text,
|
||||||
metadata_json=metadata_json or "",
|
metadata_json=metadata_json or "",
|
||||||
input_hint=input_hint,
|
input_hint=input_hint,
|
||||||
trigger=trigger,
|
trigger=trigger,
|
||||||
|
|||||||
Reference in New Issue
Block a user