Files
AIFSH-ComfyUI-DiffSynth-Studio/__init__.py
T
2024-06-27 06:13:37 +08:00

23 lines
854 B
Python

from .util_nodes import LoadVideo,PreViewVideo
from .studio_nodes import DiffTextNode,DiffutoonNode,SDPathLoader,ControlNetPathLoader
WEB_DIRECTORY = "./web"
# A dictionary that contains all nodes you want to export with their names
# NOTE: names should be globally unique
NODE_CLASS_MAPPINGS = {
"LoadVideo": LoadVideo,
"PreViewVideo": PreViewVideo,
"SDPathLoader": SDPathLoader,
"DiffTextNode": DiffTextNode,
"DiffutoonNode": DiffutoonNode,
"ControlNetPathLoader": ControlNetPathLoader
}
# A dictionary that contains the friendly/humanly readable titles for the nodes
NODE_DISPLAY_NAME_MAPPINGS = {
"LoadVideo": "LoadVideo",
"PreViewVideo": "PreViewVideo",
"SDPathLoader": "SDPathLoader",
"DiffTextNode": "DiffTextNode",
"DiffutoonNode": "DiffutoonNode",
"ControlNetPathLoader": "ControlNetPathLoader"
}