fix

2025-04-02 17:42:05 +08:00
parent 352255b216
commit d622731696
2 changed files with 15 additions and 17 deletions
@@ -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"]
@@ -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",
-}