File size: 603 Bytes
82ea528 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 |
from .nodes.model_loader import JanusModelLoader
from .nodes.image_understanding import JanusImageUnderstanding
from .nodes.image_generation import JanusImageGeneration
NODE_CLASS_MAPPINGS = {
"JanusModelLoader": JanusModelLoader,
"JanusImageUnderstanding": JanusImageUnderstanding,
"JanusImageGeneration": JanusImageGeneration,
}
NODE_DISPLAY_NAME_MAPPINGS = {
"JanusModelLoader": "Janus Model Loader",
"JanusImageUnderstanding": "Janus Image Understanding",
"JanusImageGeneration": "Janus Image Generation",
}
__all__ = ['NODE_CLASS_MAPPINGS', 'NODE_DISPLAY_NAME_MAPPINGS'] |