From fbadedc06c3a2a2637addf5b7f55c4f8c5e8b9fb Mon Sep 17 00:00:00 2001
From: comfyanonymous <121283862+comfyanonymous@users.noreply.github.com>
Date: Tue, 29 Aug 2023 18:14:47 -0400
Subject: [PATCH] Fix __init__.py

---
 __init__.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/__init__.py b/__init__.py
index ca6bb3b..c47df67 100644
--- a/__init__.py
+++ b/__init__.py
@@ -16,6 +16,7 @@ for module_name in node_list:
     imported_module = importlib.import_module(".{}".format(module_name), __name__)
 
     NODE_CLASS_MAPPINGS = {**NODE_CLASS_MAPPINGS, **imported_module.NODE_CLASS_MAPPINGS}
-    NODE_DISPLAY_NAME_MAPPINGS = {**NODE_DISPLAY_NAME_MAPPINGS, **imported_module.NODE_DISPLAY_NAME_MAPPINGS}
+    if hasattr(imported_module, "NODE_DISPLAY_NAME_MAPPINGS"):
+        NODE_DISPLAY_NAME_MAPPINGS = {**NODE_DISPLAY_NAME_MAPPINGS, **imported_module.NODE_DISPLAY_NAME_MAPPINGS}
 
 __all__ = ['NODE_CLASS_MAPPINGS', 'NODE_DISPLAY_NAME_MAPPINGS']