- PromptGeneratorNode with 7 style presets (cinematic, anime, photorealistic, fantasy, abstract, cyberpunk, sci-fi) - Temperature and Top-P sampling controls via Ollama API - Optional emphasis (focus area) and mood inputs - Reasoning toggle to show/hide model's thinking process - Jinja2 templating for customizable style templates - Graceful fallback when optional dependencies unavailable - ComfyUI Registry ready (pyproject.toml configured)
17 lines
385 B
Python
17 lines
385 B
Python
"""
|
|
ComfyUI Prompt Generator Node
|
|
Generate Stable Diffusion prompts using Qwen3-8B via Ollama
|
|
"""
|
|
|
|
from .nodes.prompt_generator_node import PromptGeneratorNode
|
|
|
|
NODE_CLASS_MAPPINGS = {
|
|
"PromptGenerator": PromptGeneratorNode,
|
|
}
|
|
|
|
NODE_DISPLAY_NAME_MAPPINGS = {
|
|
"PromptGenerator": "🎨 Prompt Generator (Qwen)",
|
|
}
|
|
|
|
__all__ = ["NODE_CLASS_MAPPINGS", "NODE_DISPLAY_NAME_MAPPINGS"]
|