| from comfy_api.latest import io
|
|
|
|
|
| RelayOptions = io.Custom("RELAY_OPTIONS")
|
|
|
|
|
|
|
| class PromptRelayAdvancedOptions(io.ComfyNode):
|
| """Per-stream temporal-penalty knobs for Prompt Relay encoders."""
|
|
|
| @classmethod
|
| def define_schema(cls):
|
| return io.Schema(
|
| node_id="PromptRelayAdvancedOptions",
|
| display_name="Prompt Relay Advanced Options",
|
| category="conditioning/prompt_relay",
|
| is_experimental=True,
|
| description=(
|
| "Optional per-stream tuning. Connect to the relay_options input on a Prompt Relay encoder. Audio fields only affect LTX2."
|
| ),
|
| inputs=[
|
| io.Float.Input(
|
| "video_strength", default=1.0, min=0.0, max=10.0, step=0.05,
|
| tooltip="Multiplier on the video temporal penalty. 0 disables video segmentation. Most useful in the 0–1 range to soften boundaries; values >1 saturate quickly at the default epsilon (1e-3) because the bias is already large enough to zero out distant tokens. To make >1 visibly meaningful, raise epsilon to ~0.1 or higher.",
|
| ),
|
| io.Float.Input(
|
| "video_window_scale", default=1.0, min=0.0, max=4.0, step=0.05,
|
| tooltip="Scales the flat anchor zone (default L/2 - 2 frames). <1 starts the soft falloff sooner; >1 widens the rigid zone. 0 collapses the anchor to a point — falloff begins immediately at the segment midpoint (sharper than default, not softer).",
|
| ),
|
| io.Float.Input(
|
| "audio_epsilon", default=0.0, min=0.0, max=0.99, step=1e-4,
|
| tooltip="Epsilon for the audio stream. 0 = inherit from the encoder's main epsilon.",
|
| ),
|
| io.Float.Input(
|
| "audio_strength", default=1.0, min=0.0, max=10.0, step=0.05,
|
| tooltip="Multiplier on the audio temporal penalty. 0 lets audio bleed across visual cuts. Most useful in the 0–1 range; values >1 saturate quickly at the default epsilon (1e-3) — raise audio_epsilon to ~0.1 or higher to make >1 visibly meaningful.",
|
| ),
|
| io.Float.Input(
|
| "audio_window_scale", default=1.0, min=0.0, max=4.0, step=0.05,
|
| tooltip="Scales the flat anchor zone width for the audio stream. <1 starts the soft falloff sooner; >1 widens the rigid zone. 0 collapses the anchor to a point — falloff begins immediately at the segment midpoint (sharper than default, not softer).",
|
| ),
|
| ],
|
| outputs=[
|
| RelayOptions.Output(display_name="relay_options"),
|
| ],
|
| )
|
|
|
| @classmethod
|
| def execute(cls, video_strength, video_window_scale,
|
| audio_epsilon, audio_strength, audio_window_scale) -> io.NodeOutput:
|
| opts = {
|
| "video_strength": video_strength,
|
| "video_window_scale": video_window_scale,
|
| "audio_epsilon": audio_epsilon if audio_epsilon > 0 else None,
|
| "audio_strength": audio_strength,
|
| "audio_window_scale": audio_window_scale,
|
| }
|
| return io.NodeOutput(opts)
|
|
|
