diff --git a/__init__.py b/__init__.py index d47bbe5..d5a3e2f 100644 --- a/__init__.py +++ b/__init__.py @@ -1,5 +1,5 @@ from .nodes import NODE_CLASS_MAPPINGS, NODE_DISPLAY_NAME_MAPPINGS -from .nodes import PromptRelayEncode, PromptRelayEncodeTimeline +from .nodes import PromptRelayEncode, PromptRelayEncodeTimeline, PromptRelayAdvancedOptions from .smart_nodes import PromptRelaySmartEncode, PromptRelaySmartEncodeTest from comfy_api.latest import ComfyExtension, io from typing_extensions import override @@ -13,6 +13,7 @@ class PromptRelay(ComfyExtension): PromptRelayEncodeTimeline, PromptRelaySmartEncode, PromptRelaySmartEncodeTest, + PromptRelayAdvancedOptions ] @@ -23,14 +24,16 @@ NODE_CLASS_MAPPINGS = { "PromptRelayEncode": PromptRelayEncode, "PromptRelayEncodeTimeline": PromptRelayEncodeTimeline, "PromptRelaySmartEncode": PromptRelaySmartEncode, - "PromptRelaySmartEncodeTest": PromptRelaySmartEncodeTest + "PromptRelaySmartEncodeTest": PromptRelaySmartEncodeTest, + "PromptRelayAdvancedOptions": PromptRelayAdvancedOptions } NODE_DISPLAY_NAME_MAPPINGS = { "PromptRelayEncode": "Prompt Relay Encode", "PromptRelayEncodeTimeline": "Prompt Relay Encode (Timeline)", "PromptRelaySmartEncode": "Prompt Relay Encode (Smart)", - "PromptRelaySmartEncodeTest": "Prompt Relay Smart Encode Test" + "PromptRelaySmartEncodeTest": "Prompt Relay Smart Encode Test", + "PromptRelayAdvancedOptions": "Prompt Relay Advanced Options" }