Update __init__.py

This commit is contained in:
kijai
2026-05-02 20:13:18 +03:00
parent e3b003c05a
commit 5cae3c0018
+6 -3
View File
@@ -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"
}