21 lines
883 B
Python
21 lines
883 B
Python
from .ComfyMflux import QuickMfluxNode, MfluxModelsLoader, MfluxModelsDownloader, MfluxCustomModels, MfluxLoadImg2Img, MfluxLorasLoader, MfluxControlNetLoader
|
|
|
|
NODE_CLASS_MAPPINGS = {
|
|
"QuickMfluxNode": QuickMfluxNode,
|
|
"MfluxModelsLoader": MfluxModelsLoader,
|
|
"MfluxModelsDownloader": MfluxModelsDownloader,
|
|
"MfluxCustomModels": MfluxCustomModels,
|
|
"MfluxLoadImg2Img": MfluxLoadImg2Img,
|
|
"MfluxLorasLoader": MfluxLorasLoader,
|
|
"MfluxControlNetLoader": MfluxControlNetLoader,
|
|
}
|
|
|
|
NODE_DISPLAY_NAME_MAPPINGS = {
|
|
"QuickMfluxNode": "Quick MFlux Generation",
|
|
"MfluxModelsLoader": "MFlux Models Loader",
|
|
"MfluxModelsDownloader": "MFlux Models Downloader",
|
|
"MfluxCustomModels": "MFlux Custom Models",
|
|
"MfluxLoadImg2Img": "MFlux Img2Img Loader",
|
|
"MfluxLorasLoader": "MFlux Loras Loader",
|
|
"MfluxControlNetLoader": "MFlux ControlNet Loader",
|
|
} |