From 26d299de69baac8a8a0f8f65472bbad1b08f5d10 Mon Sep 17 00:00:00 2001 From: Pheat-AI Date: Thu, 5 Sep 2024 21:55:49 -0700 Subject: [PATCH] Update __init__.py Added new overlay node --- __init__.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/__init__.py b/__init__.py index f47fa7e..53c1fcc 100644 --- a/__init__.py +++ b/__init__.py @@ -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']