14 lines
510 B
Python
14 lines
510 B
Python
from .prompt_with_image import PromptWithImage # Import your actual node class
|
|
|
|
NODE_CLASS_MAPPINGS = {
|
|
"PromptWithImage": PromptWithImage # The key is what will show up in ComfyUI interface
|
|
}
|
|
|
|
# Optional: If your node has display names different from class names
|
|
NODE_DISPLAY_NAME_MAPPINGS = {
|
|
"PromptWithImage": "LLM Chat (Image Optional)" # Optional, for better UI display
|
|
}
|
|
|
|
# Required: To properly register your node
|
|
__all__ = ['NODE_CLASS_MAPPINGS', 'NODE_DISPLAY_NAME_MAPPINGS']
|