41 lines
1.3 KiB
Python
41 lines
1.3 KiB
Python
"""Top-level package for openoutpaint_comfyui_interface."""
|
|
|
|
__all__ = [
|
|
"NODE_CLASS_MAPPINGS",
|
|
"NODE_DISPLAY_NAME_MAPPINGS",
|
|
"WEB_DIRECTORY",
|
|
]
|
|
|
|
__author__ = """Without Ordinary"""
|
|
__email__ = "without-ordinary@users.noreply.github.com"
|
|
__version__ = "0.0.1"
|
|
|
|
WEB_DIRECTORY = "./web"
|
|
|
|
# from .py.nodes import register_nodes as openoutpaint_serving_register_nodes
|
|
# NODE_CLASS_MAPPINGS, NODE_DISPLAY_NAME_MAPPINGS = openoutpaint_serving_register_nodes()
|
|
|
|
NODE_CLASS_MAPPINGS = {}
|
|
NODE_DISPLAY_NAME_MAPPINGS = {}
|
|
def add_nodes(node_classes):
|
|
for node_class in node_classes:
|
|
NODE_CLASS_MAPPINGS[node_class.CLASSNAME] = node_class
|
|
NODE_DISPLAY_NAME_MAPPINGS[node_class.CLASSNAME] = node_class.NAME
|
|
|
|
from .py.nodes_serving import get_nodes as get_nodes_serving
|
|
from .py.nodes_upscale import get_nodes as get_nodes_upscale
|
|
from .py.nodes_interrogate import get_nodes as get_nodes_interrogate
|
|
from .py.nodes_txt2img import get_nodes as get_nodes_txt2img
|
|
from .py.nodes_img2img import get_nodes as get_nodes_img2img
|
|
from .py.nodes_style import get_nodes as get_nodes_style
|
|
from .py.nodes_model import get_nodes as get_nodes_model
|
|
|
|
add_nodes(get_nodes_serving())
|
|
add_nodes(get_nodes_upscale())
|
|
add_nodes(get_nodes_interrogate())
|
|
add_nodes(get_nodes_txt2img())
|
|
add_nodes(get_nodes_img2img())
|
|
add_nodes(get_nodes_style())
|
|
add_nodes(get_nodes_model())
|
|
|