26 lines
964 B
Python
26 lines
964 B
Python
from .runpod_nodes import RunPodImageNodeSDXL, RunPodImageNodeSD, RunPodImageNodeComfy, RunPodVariablesBuilderNode
|
|
from .runpod_public import PUBLIC_NODE_CLASS_MAPPINGS, PUBLIC_NODE_DISPLAY_NAME_MAPPINGS
|
|
|
|
NODE_CLASS_MAPPINGS = {
|
|
# Custom endpoint nodes
|
|
"RunPodImageNodeSDXL": RunPodImageNodeSDXL,
|
|
"RunPodImageNodeSD": RunPodImageNodeSD,
|
|
"RunPodImageNodeComfy": RunPodImageNodeComfy,
|
|
"RunPodVariablesBuilderNode": RunPodVariablesBuilderNode,
|
|
}
|
|
|
|
# Add all public endpoint nodes
|
|
NODE_CLASS_MAPPINGS.update(PUBLIC_NODE_CLASS_MAPPINGS)
|
|
|
|
NODE_DISPLAY_NAME_MAPPINGS = {
|
|
# Custom endpoint nodes
|
|
"RunPodImageNodeSDXL": "RunPod SDXL (Custom Endpoint)",
|
|
"RunPodImageNodeSD": "RunPod SD (Custom Endpoint)",
|
|
"RunPodImageNodeComfy": "RunPod ComfyUI Workflow Runner",
|
|
"RunPodVariablesBuilderNode": "RunPod Variables Builder",
|
|
}
|
|
|
|
# Add all public endpoint display names
|
|
NODE_DISPLAY_NAME_MAPPINGS.update(PUBLIC_NODE_DISPLAY_NAME_MAPPINGS)
|
|
|