32 lines
1.0 KiB
Python
32 lines
1.0 KiB
Python
from .utilities import (
|
|
DistributedSeed,
|
|
DistributedModelName,
|
|
ImageBatchDivider,
|
|
AudioBatchDivider,
|
|
DistributedEmptyImage,
|
|
AnyType,
|
|
ByPassTypeTuple,
|
|
any_type,
|
|
)
|
|
from .collector import DistributedCollectorNode
|
|
from .queue_node import DistributedQueueNode
|
|
|
|
NODE_CLASS_MAPPINGS = {
|
|
"DistributedQueue": DistributedQueueNode,
|
|
"DistributedCollector": DistributedCollectorNode,
|
|
"DistributedSeed": DistributedSeed,
|
|
"DistributedModelName": DistributedModelName,
|
|
"ImageBatchDivider": ImageBatchDivider,
|
|
"AudioBatchDivider": AudioBatchDivider,
|
|
"DistributedEmptyImage": DistributedEmptyImage,
|
|
}
|
|
NODE_DISPLAY_NAME_MAPPINGS = {
|
|
"DistributedQueue": "Distributed Queue",
|
|
"DistributedCollector": "Distributed Collector",
|
|
"DistributedSeed": "Distributed Seed",
|
|
"DistributedModelName": "Distributed Model Name",
|
|
"ImageBatchDivider": "Image Batch Divider",
|
|
"AudioBatchDivider": "Audio Batch Divider",
|
|
"DistributedEmptyImage": "Distributed Empty Image",
|
|
}
|