File size: 3,283 Bytes
7cc9dda | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 | 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)
|