32 lines
987 B
Python
32 lines
987 B
Python
from .nodes.height_node import HeightNode
|
|
from .nodes.random_value_tracker import SeedHistory
|
|
from .nodes.sampler_selector import SamplerSelector
|
|
from .nodes.scheduler_selector import SchedulerSelector
|
|
from .nodes.width_height_node import WidthHeightNode
|
|
from .nodes.width_node import WidthNode
|
|
|
|
# Print startup message in blue
|
|
print("\033[94m[ComfyAssets Selectors] Loaded..\033[0m")
|
|
|
|
WEB_DIRECTORY = "./web"
|
|
|
|
NODE_CLASS_MAPPINGS = {
|
|
"SamplerSelector": SamplerSelector,
|
|
"SchedulerSelector": SchedulerSelector,
|
|
"SeedHistory": SeedHistory,
|
|
"WidthNode": WidthNode,
|
|
"HeightNode": HeightNode,
|
|
"WidthHeightNode": WidthHeightNode,
|
|
}
|
|
|
|
NODE_DISPLAY_NAME_MAPPINGS = {
|
|
"SamplerSelector": "Sampler Selector",
|
|
"SchedulerSelector": "Scheduler Selector",
|
|
"SeedHistory": "Seed History",
|
|
"WidthNode": "Width",
|
|
"HeightNode": "Height",
|
|
"WidthHeightNode": "Width & Height",
|
|
}
|
|
|
|
__all__ = ["NODE_CLASS_MAPPINGS", "NODE_DISPLAY_NAME_MAPPINGS", "WEB_DIRECTORY"]
|