38 lines
1.5 KiB
Python
38 lines
1.5 KiB
Python
from .loader_nodes import FolderImageLoaderNode
|
|
from .index_selector_nodes import IndexedImageSelectorNode, IndexedNameSelectorNode
|
|
from .save_nodes import SaveExactNameImageNode
|
|
from .logic_nodes import IsLastIndexNode
|
|
from .aws_event_node import EventBridgeTriggerNode
|
|
from .task_manager import TaskIDStorageNode
|
|
from .load_pre_models import (
|
|
StringViewer,
|
|
LoadPreCheckpointModel,
|
|
LoadPreControlNetModel,
|
|
)
|
|
|
|
NODE_CLASS_MAPPINGS = {
|
|
"FolderImageLoaderNode": FolderImageLoaderNode,
|
|
"IndexedImageSelectorNode": IndexedImageSelectorNode,
|
|
"IndexedNameSelectorNode": IndexedNameSelectorNode,
|
|
"SaveExactNameImageNode": SaveExactNameImageNode,
|
|
"IsLastIndexNode": IsLastIndexNode,
|
|
"EventBridgeTriggerNode": EventBridgeTriggerNode,
|
|
"TaskIDStorageNode": TaskIDStorageNode,
|
|
"StringViewer": StringViewer,
|
|
"LoadPreCheckpointModel": LoadPreCheckpointModel,
|
|
"LoadPreControlNetModel": LoadPreControlNetModel,
|
|
}
|
|
|
|
NODE_DISPLAY_NAME_MAPPINGS = {
|
|
"FolderImageLoaderNode": "Load Images from Folder",
|
|
"IndexedImageSelectorNode": "Select Image by Index",
|
|
"IndexedNameSelectorNode": "Get Filename by Index",
|
|
"SaveExactNameImageNode": "Save Image with Exact Filename",
|
|
"IsLastIndexNode": "Check If Last Index",
|
|
"EventBridgeTriggerNode": "Trigger EventBridge (Simulated)",
|
|
"TaskIDStorageNode": "Get TaskID",
|
|
"StringViewer": "Simple String View",
|
|
"LoadPreCheckpointModel": "Load Pre Checkpoint Model",
|
|
"LoadPreControlNetModel": "Load Pre ControlNet Model",
|
|
}
|