18 lines
503 B
Python
18 lines
503 B
Python
from .nodes.model_selector import ModelSelector
|
|
from .nodes.model_loader import ModelLoader
|
|
from .nodes.size_selector import ModelSizeSelector
|
|
|
|
NODE_CLASS_MAPPINGS = {
|
|
"ModelSelector": ModelSelector,
|
|
"ModelLoader": ModelLoader,
|
|
"ModelSizeSelector": ModelSizeSelector,
|
|
}
|
|
|
|
NODE_DISPLAY_NAME_MAPPINGS = {
|
|
"ModelSelector": "Model Selector",
|
|
"ModelLoader": "Model Loader",
|
|
"ModelSizeSelector": "Model Size Selector",
|
|
}
|
|
|
|
__all__ = ["NODE_CLASS_MAPPINGS", "NODE_DISPLAY_NAME_MAPPINGS"]
|