41 lines
1.4 KiB
Python
41 lines
1.4 KiB
Python
from .nodes_ollama_router_new import (
|
|
OllamaPromptPlanner,
|
|
OllamaVisionStylePlanner,
|
|
DynamicCheckpointLoader,
|
|
DynamicLoraStack,
|
|
RouteConditioningByType,
|
|
OptionalControlNetApply,
|
|
RouteLatentByBool,
|
|
RouteVaeByBool,
|
|
OllamaDebugInfo,
|
|
OllamaRegistryInfo,
|
|
)
|
|
|
|
NODE_CLASS_MAPPINGS = {
|
|
"OllamaPromptPlanner": OllamaPromptPlanner,
|
|
"OllamaVisionStylePlanner": OllamaVisionStylePlanner,
|
|
"DynamicCheckpointLoader": DynamicCheckpointLoader,
|
|
"DynamicLoraStack": DynamicLoraStack,
|
|
"RouteConditioningByType": RouteConditioningByType,
|
|
"OptionalControlNetApply": OptionalControlNetApply,
|
|
"RouteLatentByBool": RouteLatentByBool,
|
|
"RouteVaeByBool": RouteVaeByBool,
|
|
"OllamaDebugInfo": OllamaDebugInfo,
|
|
"OllamaRegistryInfo": OllamaRegistryInfo,
|
|
}
|
|
|
|
NODE_DISPLAY_NAME_MAPPINGS = {
|
|
"OllamaPromptPlanner": "Ollama Prompt Planner",
|
|
"OllamaVisionStylePlanner": "Ollama Vision Style Planner",
|
|
"DynamicCheckpointLoader": "Dynamic Checkpoint Loader (Universal)",
|
|
"DynamicLoraStack": "Dynamic LoRA Stack",
|
|
"RouteConditioningByType": "Route Conditioning (SDXL/SD1.5)",
|
|
"OptionalControlNetApply": "Optional ControlNet Apply",
|
|
"RouteLatentByBool": "Route Latent (Bool)",
|
|
"RouteVaeByBool": "Route VAE (Bool)",
|
|
"OllamaDebugInfo": "Ollama Plan Debug Info",
|
|
"OllamaRegistryInfo": "Ollama Registry Debug Info",
|
|
}
|
|
|
|
__all__ = ["NODE_CLASS_MAPPINGS", "NODE_DISPLAY_NAME_MAPPINGS"]
|