File size: 614 Bytes
9beaf60 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 |
# only import if running as a custom node
from .nodes.lora import NunchakuFluxLoraLoader
from .nodes.models import NunchakuFluxDiTLoader, NunchakuTextEncoderLoader
from .nodes.preprocessors import FluxDepthPreprocessor
NODE_CLASS_MAPPINGS = {
"NunchakuFluxDiTLoader": NunchakuFluxDiTLoader,
"NunchakuTextEncoderLoader": NunchakuTextEncoderLoader,
"NunchakuFluxLoraLoader": NunchakuFluxLoraLoader,
"NunchakuDepthPreprocessor": FluxDepthPreprocessor,
}
NODE_DISPLAY_NAME_MAPPINGS = {k: v.TITLE for k, v in NODE_CLASS_MAPPINGS.items()}
__all__ = ["NODE_CLASS_MAPPINGS", "NODE_DISPLAY_NAME_MAPPINGS"]
|