-
Notifications
You must be signed in to change notification settings - Fork 4
/
Ostris_Nodes.py
46 lines (41 loc) · 1.29 KB
/
Ostris_Nodes.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
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
from .utils.paths import fix_import_paths
fix_import_paths()
from .nodes.general import OstrisOneSeedNode, OstrisTextBoxNode
from .nodes.llm import OstrisLLMPipeLoader, OstrisPromptUpsampler
from .nodes.batch_image_loader import OstrisBatchImageLoader
ostris_node_list = [
{
"uid": "One Seed - Ostris",
"class": OstrisOneSeedNode,
"title": "One Seed",
},
{
"uid": "Text Box - Ostris",
"class": OstrisTextBoxNode,
"title": "Text Box",
},
{
"uid": "LLM Pipe Loader - Ostris",
"class": OstrisLLMPipeLoader,
"title": "LLM Pipe Loader",
},
{
"uid": "LLM Prompt Upsampling - Ostris",
"class": OstrisPromptUpsampler,
"title": "Prompt Upsampling",
},
# {
# "uid": "Batch Image Loader - Ostris",
# "class": OstrisBatchImageLoader,
# "title": "Text Box",
# }
]
# A dictionary that contains all nodes you want to export with their names
# NOTE: names should be globally unique
NODE_CLASS_MAPPINGS = {}
# A dictionary that contains the friendly/humanly readable titles for the nodes
NODE_DISPLAY_NAME_MAPPINGS = {}
# build the mappings
for n in ostris_node_list:
NODE_CLASS_MAPPINGS[n["uid"]] = n["class"]
NODE_DISPLAY_NAME_MAPPINGS[n["uid"]] = f"{n['title']}_uid"