-
Notifications
You must be signed in to change notification settings - Fork 11
/
Copy path__init__.py
26 lines (25 loc) · 966 Bytes
/
__init__.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
from .nodes_legacy import MiniCPM_VQA
from .nodes_polished import MiniCPM_VQA_Polished
from .image_nodes import MultipleImagesInput
from .util_nodes import LoadVideo,PreviewVideo
from .display_text_nodes import DisplayText
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,
"MultipleImagesInput": MultipleImagesInput,
"MiniCPM_VQA": MiniCPM_VQA,
"MiniCPM_VQA_Polished": MiniCPM_VQA_Polished,
"DisplayText": DisplayText,
}
# A dictionary that contains the friendly/humanly readable titles for the nodes
NODE_DISPLAY_NAME_MAPPINGS = {
"LoadVideo": "Load Video",
"PreviewVideo": "Preview Video",
"MultipleImagesInput": "Multiple Images Input",
"MiniCPM_VQA": "MiniCPM VQA",
"MiniCPM_VQA_Polished": "MiniCPM VQA Polished",
"DisplayText": "Display Text",
}