Files
Pheat-AI-Remade_nodes/__init__.py
T
2024-09-05 22:22:03 -07:00

10 lines
721 B
Python

from .batch_image_blend import NODE_CLASS_MAPPINGS as BLEND_CLASS_MAPPINGS, NODE_DISPLAY_NAME_MAPPINGS as BLEND_DISPLAY_NAMES
from .batch_image_overlay import NODE_CLASS_MAPPINGS as OVERLAY_CLASS_MAPPINGS, NODE_DISPLAY_NAME_MAPPINGS as OVERLAY_DISPLAY_NAMES
from .remove_black_to_transparent import NODE_CLASS_MAPPINGS as REMOVE_BLACK_CLASS_MAPPINGS, NODE_DISPLAY_NAME_MAPPINGS as REMOVE_BLACK_DISPLAY_NAMES
# Merge all node mappings into one
NODE_CLASS_MAPPINGS = {**BLEND_CLASS_MAPPINGS, **OVERLAY_CLASS_MAPPINGS, **REMOVE_BLACK_CLASS_MAPPINGS}
NODE_DISPLAY_NAME_MAPPINGS = {**BLEND_DISPLAY_NAMES, **OVERLAY_DISPLAY_NAMES, **REMOVE_BLACK_DISPLAY_NAMES}
__all__ = ['NODE_CLASS_MAPPINGS', 'NODE_DISPLAY_NAME_MAPPINGS']