20 lines
520 B
Python
20 lines
520 B
Python
"""
|
|
ComfyUI Prompt Generator Node
|
|
Generate Stable Diffusion prompts using Qwen3-8B via Ollama
|
|
"""
|
|
|
|
from .nodes.prompt_generator_node import PromptGeneratorNode
|
|
from .nodes.style_applier_node import StyleApplierNode
|
|
|
|
NODE_CLASS_MAPPINGS = {
|
|
"PromptGenerator": PromptGeneratorNode,
|
|
"StyleApplier": StyleApplierNode,
|
|
}
|
|
|
|
NODE_DISPLAY_NAME_MAPPINGS = {
|
|
"PromptGenerator": "🎨 Prompt Generator (Qwen)",
|
|
"StyleApplier": "🎬 Style Applier",
|
|
}
|
|
|
|
__all__ = ["NODE_CLASS_MAPPINGS", "NODE_DISPLAY_NAME_MAPPINGS"]
|