poxularandCursor
d101418eb9
Keeps CPU parity nodes and providers unchanged; CUDA modules and ComfyUI nodes run alongside with CPU fallback when CUDA EP is unavailable. Co-authored-by: Cursor <cursoragent@cursor.com>
23 lines
798 B
Python
23 lines
798 B
Python
"""
|
|
MAGICMATCH — ComfyUI custom nodes for neural color match (ONNX).
|
|
"""
|
|
|
|
import os
|
|
|
|
from .nodes import NODE_CLASS_MAPPINGS, NODE_DISPLAY_NAME_MAPPINGS
|
|
|
|
# Experimental CUDA nodes (distinct class names; safe alongside CPU nodes).
|
|
# Set MAGICMATCH_CUDA_NODES=0 to hide them from ComfyUI.
|
|
if os.environ.get("MAGICMATCH_CUDA_NODES", "1") != "0":
|
|
from .nodes_cuda import CUDA_NODE_CLASS_MAPPINGS, CUDA_NODE_DISPLAY_NAME_MAPPINGS
|
|
|
|
NODE_CLASS_MAPPINGS = {**NODE_CLASS_MAPPINGS, **CUDA_NODE_CLASS_MAPPINGS}
|
|
NODE_DISPLAY_NAME_MAPPINGS = {
|
|
**NODE_DISPLAY_NAME_MAPPINGS,
|
|
**CUDA_NODE_DISPLAY_NAME_MAPPINGS,
|
|
}
|
|
|
|
WEB_DIRECTORY = os.path.join(os.path.dirname(__file__), "js")
|
|
|
|
__all__ = ["NODE_CLASS_MAPPINGS", "NODE_DISPLAY_NAME_MAPPINGS", "WEB_DIRECTORY"]
|