17 lines
510 B
Python
17 lines
510 B
Python
from .flux_prompt_generator_node import FluxPromptGeneratorNode
|
|
from .flux_image_caption_node import FluxImageCaptionNode
|
|
|
|
NODE_CLASS_MAPPINGS = {
|
|
"FluxPromptGeneratorNode": FluxPromptGeneratorNode,
|
|
"FluxImageCaptionNode": FluxImageCaptionNode
|
|
}
|
|
|
|
NODE_DISPLAY_NAME_MAPPINGS = {
|
|
"FluxPromptGeneratorNode": "Flux Prompt Generator",
|
|
"FluxImageCaptionNode": "Flux Image Caption"
|
|
}
|
|
|
|
__all__ = ['NODE_CLASS_MAPPINGS', 'NODE_DISPLAY_NAME_MAPPINGS']
|
|
|
|
print("ComfyUI_FluxPromptGen loaded successfully")
|