29 lines
901 B
Python
29 lines
901 B
Python
WEB_DIRECTORY = "./web"
|
|
|
|
from .nodes.height_node import HeightNode
|
|
from .nodes.sampler_selector import SamplerSelector
|
|
from .nodes.scheduler_selector import SchedulerSelector
|
|
from .nodes.seed_generator import SeedGenerator
|
|
from .nodes.width_height_node import WidthHeightNode
|
|
from .nodes.width_node import WidthNode
|
|
|
|
NODE_CLASS_MAPPINGS = {
|
|
"SamplerSelector": SamplerSelector,
|
|
"SchedulerSelector": SchedulerSelector,
|
|
"SeedGenerator": SeedGenerator,
|
|
"WidthNode": WidthNode,
|
|
"HeightNode": HeightNode,
|
|
"WidthHeightNode": WidthHeightNode,
|
|
}
|
|
|
|
NODE_DISPLAY_NAME_MAPPINGS = {
|
|
"SamplerSelector": "Sampler Selector",
|
|
"SchedulerSelector": "Scheduler Selector",
|
|
"SeedGenerator": "Seed Generator",
|
|
"WidthNode": "Width",
|
|
"HeightNode": "Height",
|
|
"WidthHeightNode": "Width & Height",
|
|
}
|
|
|
|
__all__ = ["NODE_CLASS_MAPPINGS", "NODE_DISPLAY_NAME_MAPPINGS", "WEB_DIRECTORY"]
|