from .nodes.load_dilight_controlnet_node import LoadDiLightControlNetNode
from .nodes.prepare_dilight_cond_node import PrepareDiLightCondNode


NODE_CLASS_MAPPINGS = {
    "LoadDiLightControlNet": LoadDiLightControlNetNode,
    "PrepareDiLightCond": PrepareDiLightCondNode,
}

NODE_DISPLAY_NAME_MAPPINGS = {
    "LoadDiLightControlNet": "Load DiLight ControlNet",
    "PrepareDiLightCond": "Prepare DiLight CN Image",
}