Update __init__.py

Added new overlay node
This commit is contained in:
Pheat-AI
2024-09-05 21:55:49 -07:00
committed by GitHub
parent 8c68c1264c
commit 26d299de69
+6 -1
View File
@@ -1,3 +1,8 @@
from .batch_image_blend import NODE_CLASS_MAPPINGS, NODE_DISPLAY_NAME_MAPPINGS
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
# Merge both node mappings into one
NODE_CLASS_MAPPINGS = {**BLEND_CLASS_MAPPINGS, **OVERLAY_CLASS_MAPPINGS}
NODE_DISPLAY_NAME_MAPPINGS = {**BLEND_DISPLAY_NAMES, **OVERLAY_DISPLAY_NAMES}
__all__ = ['NODE_CLASS_MAPPINGS', 'NODE_DISPLAY_NAME_MAPPINGS']