Update __init__.py
This commit is contained in:
+6
-3
@@ -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"
|
||||
}
|
||||
|
||||
|
||||
|
||||
Reference in New Issue
Block a user