16 lines
460 B
Python
16 lines
460 B
Python
try:
|
|
import comfy.utils
|
|
except ImportError:
|
|
pass
|
|
else:
|
|
NODE_CLASS_MAPPINGS = {}
|
|
|
|
from .Text.nodes import NODE_CLASS_MAPPINGS as Text_Nodes
|
|
NODE_CLASS_MAPPINGS.update(Text_Nodes)
|
|
|
|
from .Image.nodes import NODE_CLASS_MAPPINGS as Image_Nodes
|
|
NODE_CLASS_MAPPINGS.update(Image_Nodes)
|
|
|
|
NODE_DISPLAY_NAME_MAPPINGS = {k:v.TITLE for k,v in NODE_CLASS_MAPPINGS.items()}
|
|
|
|
__all__ = ['NODE_CLASS_MAPPINGS', 'NODE_DISPLAY_NAME_MAPPINGS'] |