import custom_nodes.comfy_nodes_trojblue.image_layering as image_layering
import custom_nodes.comfy_nodes_trojblue.color_correction as color_correction
import custom_nodes.comfy_nodes_trojblue.model_router as model_router



NODE_CLASS_MAPPINGS = {
    "trLayering": image_layering.Layering,                      # Layering
    "trColorCorrection": color_correction.ColorCorrectionNode,  # ColorCorrectionNode
    "trRouter": model_router.ModelRouterPlugin,          # ModelRouterPlugin

}