From d6227316965e9d49400a4f83fbfe4d7d79ea832e Mon Sep 17 00:00:00 2001
From: jax <jaxinformationtransformer@gmail.com>
Date: Wed, 2 Apr 2025 17:42:05 +0800
Subject: [PATCH] fix

---
 __init__.py          | 18 +++++++++++++++---
 nodes/comfy_nodes.py | 14 --------------
 2 files changed, 15 insertions(+), 17 deletions(-)

diff --git a/__init__.py b/__init__.py
index e54725e..46c9450 100644
--- a/__init__.py
+++ b/__init__.py
@@ -1,9 +1,21 @@
 
-import nodes.comfy_nodes
+from .nodes.comfy_nodes import EasyControlLoadFlux, EasyControlLoadLora, EasyControlLoadMultiLora, EasyControlGenerate
 
-NODE_CLASS_MAPPINGS = nodes.comfy_nodes.NODE_CLASS_MAPPINGS
-NODE_DISPLAY_NAME_MAPPINGS = nodes.comfy_nodes.NODE_DISPLAY_NAME_MAPPINGS
 
+# 注册节点
+NODE_CLASS_MAPPINGS = {
+    "EasyControlLoadFlux": EasyControlLoadFlux,
+    "EasyControlLoadLora": EasyControlLoadLora,
+    "EasyControlLoadMultiLora": EasyControlLoadMultiLora,
+    "EasyControlGenerate": EasyControlGenerate,
+}
+
+NODE_DISPLAY_NAME_MAPPINGS = {
+    "EasyControlLoadFlux": "Load EasyControl Flux Model",
+    "EasyControlLoadLora": "Load EasyControl Lora",
+    "EasyControlLoadMultiLora": "Load Multiple EasyControl Loras",
+    "EasyControlGenerate": "EasyControl Generate",
+} 
 WEB_DIRECTORY = "./web"
 
 __all__ = ['NODE_CLASS_MAPPINGS', 'NODE_DISPLAY_NAME_MAPPINGS', "WEB_DIRECTORY"]
\ No newline at end of file
diff --git a/nodes/comfy_nodes.py b/nodes/comfy_nodes.py
index 8b63990..2f69aa4 100644
--- a/nodes/comfy_nodes.py
+++ b/nodes/comfy_nodes.py
@@ -167,17 +167,3 @@ class EasyControlGenerate:
         
         return (image,)
 
-# 注册节点
-NODE_CLASS_MAPPINGS = {
-    "EasyControlLoadFlux": EasyControlLoadFlux,
-    "EasyControlLoadLora": EasyControlLoadLora,
-    "EasyControlLoadMultiLora": EasyControlLoadMultiLora,
-    "EasyControlGenerate": EasyControlGenerate,
-}
-
-NODE_DISPLAY_NAME_MAPPINGS = {
-    "EasyControlLoadFlux": "Load EasyControl Flux Model",
-    "EasyControlLoadLora": "Load EasyControl Lora",
-    "EasyControlLoadMultiLora": "Load Multiple EasyControl Loras",
-    "EasyControlGenerate": "EasyControl Generate",
-} 
\ No newline at end of file