Files
2025-02-13 07:59:58 -05:00

25 lines
941 B
Python

from .nodes import EncodedPromptToFile, EncodedPromptFromFile, SampledLatentsToFile, SampledLatentsFromFile, IO_LoadImage
# Map all your custom nodes classes with the names that will be displayed in the UI.
NODE_CLASS_MAPPINGS = {
"EncodedPromptToFile": EncodedPromptToFile,
"EncodedPromptFromFile": EncodedPromptFromFile,
"SampledLatentsToFile": SampledLatentsToFile,
"SampledLatentsFromFile": SampledLatentsFromFile,
"IO_LoadImage": IO_LoadImage,
}
NODE_DISPLAY_NAMES = {
"EncodedPromptToFile": "Encoded Prompt [to file] ►►► 💾",
"EncodedPromptFromFile": "Encoded Prompt [from file] ◄◄◄ 📁",
"SampledLatentsToFile": "Sampled Latents [to file] ►►► 💾",
"SampledLatentsFromFile": "Sampled Latents [from file] ◄◄◄ 📁",
"IO_LoadImage": "Load Image ◄◄◄ 📁",
}
WEB_DIRECTORY = "./web"
__all__ = ['NODE_CLASS_MAPPINGS', 'NODE_DISPLAY_NAMES', 'WEB_DIRECTORY']