nodes: add cable management node

README.md

@@ -22,7 +22,8 @@ image_layering:
 color_correction:
 - Adjusts the color of the target image according to another image; ported from stable diffusion WebUI
 
-
+model_router:
+- Batch reroutes `model`, `clip`, `vae`, `condition_1` and `condition_2` for cleaner workflow
 
 ## External Nodes
 

__init__.py

@@ -5,8 +5,8 @@ import custom_nodes.comfy_nodes_trojblue.model_router as model_router
 
 
 NODE_CLASS_MAPPINGS = {
-    "layering": image_layering.Layering,                      # Layering
-    "color_correction": color_correction.ColorCorrectionNode,  # ColorCorrectionNode
+    "trLayering": image_layering.Layering,                      # Layering
+    "trColorCorrection": color_correction.ColorCorrectionNode,  # ColorCorrectionNode
     "trRouter": model_router.ModelRouterPlugin,          # ModelRouterPlugin
 
 }

color_correction.py

@@ -57,6 +57,3 @@ class ColorCorrectionNode:
 
         return (out_image,)
 
-NODE_CLASS_MAPPINGS = {
-    "ColorCorrectionNode": ColorCorrectionNode
-}

image_layering.py

@@ -90,9 +90,6 @@ class Layering:
 
         return (out_image,)
 
-NODE_CLASS_MAPPINGS = {
-    "Layering": Layering,
-}
 
 
 

model_router.py

@@ -43,9 +43,3 @@ class ModelRouterPlugin:
     def execute(self, model=None, clip=None, vae=None, conditioning1=None, conditioning2=None):
         return model, clip, vae, conditioning1, conditioning2
 
-
-# A dictionary that contains all nodes you want to export with their names
-# NOTE: names should be globally unique
-NODE_CLASS_MAPPINGS = {
-    "trRouter": ModelRouterPlugin
-}