Files
raysers-Mflux-ComfyUI/__init__.py
T
2024-10-23 22:55:05 -07:00

20 lines
796 B
Python

from .Mflux_Air import QuickMfluxNode, MfluxModelsLoader, MfluxModelsDownloader, MfluxCustomModels
from .Mflux_Pro import MfluxLorasLoader, MfluxControlNetLoader
NODE_CLASS_MAPPINGS = {
"QuickMfluxNode": QuickMfluxNode,
"MfluxModelsLoader": MfluxModelsLoader,
"MfluxModelsDownloader": MfluxModelsDownloader,
"MfluxCustomModels": MfluxCustomModels,
"MfluxLorasLoader": MfluxLorasLoader,
"MfluxControlNetLoader": MfluxControlNetLoader,
}
NODE_DISPLAY_NAME_MAPPINGS = {
"QuickMfluxNode": "Quick MFlux Generation",
"MfluxModelsLoader": "Mflux Models Loader",
"MfluxModelsDownloader": "Mflux Models Downloader",
"MfluxCustomModels": "Mflux Custom Models",
"MfluxLorasLoader": "Mflux Loras Loader",
"MfluxControlNetLoader": "Mflux ControlNet Loader",
}