22 lines
635 B
Python
22 lines
635 B
Python
try:
|
|
from .py.plugin import load_node_classes, register_gallery_routes
|
|
except ImportError:
|
|
from py.plugin import load_node_classes, register_gallery_routes
|
|
|
|
|
|
_NODE_CLASSES = load_node_classes()
|
|
UniversalJsonSegmentRandomizer = _NODE_CLASSES["UniversalJsonSegmentRandomizer"]
|
|
register_gallery_routes()
|
|
|
|
|
|
NODE_CLASS_MAPPINGS = dict(_NODE_CLASSES)
|
|
|
|
NODE_DISPLAY_NAME_MAPPINGS = {
|
|
"UniversalJsonSegmentRandomizer": "Universal Artist/Tag Randomizer",
|
|
"UniversalPromptSnapshot": "Universal Prompt Snapshot",
|
|
}
|
|
|
|
WEB_DIRECTORY = "./web/comfyui"
|
|
|
|
__all__ = ["NODE_CLASS_MAPPINGS", "NODE_DISPLAY_NAME_MAPPINGS", "WEB_DIRECTORY"]
|