17 lines
649 B
Python
17 lines
649 B
Python
from .prompt_loop_node import NODE_CLASS_MAPPINGS as PROMPT_NODE_CLASS_MAPPINGS
|
|
from .prompt_loop_node import NODE_DISPLAY_NAME_MAPPINGS as PROMPT_NODE_DISPLAY_NAME_MAPPINGS
|
|
from .batch_image_loader import NODE_CLASS_MAPPINGS as IMAGE_NODE_CLASS_MAPPINGS
|
|
from .batch_image_loader import NODE_DISPLAY_NAME_MAPPINGS as IMAGE_NODE_DISPLAY_NAME_MAPPINGS
|
|
|
|
# Combine all node mappings
|
|
NODE_CLASS_MAPPINGS = {
|
|
**PROMPT_NODE_CLASS_MAPPINGS,
|
|
**IMAGE_NODE_CLASS_MAPPINGS
|
|
}
|
|
|
|
NODE_DISPLAY_NAME_MAPPINGS = {
|
|
**PROMPT_NODE_DISPLAY_NAME_MAPPINGS,
|
|
**IMAGE_NODE_DISPLAY_NAME_MAPPINGS
|
|
}
|
|
|
|
__all__ = ['NODE_CLASS_MAPPINGS', 'NODE_DISPLAY_NAME_MAPPINGS'] |