Skip to content

GitLab

Switch branch/tag
Find file Normal viewHistoryPermalink
__init__.py 1.18 KB
Newer Older
raojy's avatar
fix  
raojy committed
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36 
"""
ComfyUI Custom Nodes for LLaDA2.0_Uni
Unified multimodal: Text-to-Image, Image Understanding (VQA), Image Editing

This node package lives inside LLaDA2.0_Uni/apps/comfyui/ and imports
encoder/decoder from the parent project directly.
"""

from .nodes import (
    LLaDA2UniLoader,
    LLaDA2UniTextToImage,
    LLaDA2UniImageUnderstanding,
    LLaDA2UniImageEditing,
    LLaDA2UniImageDecode,
    LLaDA2UniUnloadModel,
)

NODE_CLASS_MAPPINGS = {
    "LLaDA2UniLoader": LLaDA2UniLoader,
    "LLaDA2UniTextToImage": LLaDA2UniTextToImage,
    "LLaDA2UniImageUnderstanding": LLaDA2UniImageUnderstanding,
    "LLaDA2UniImageEditing": LLaDA2UniImageEditing,
    "LLaDA2UniImageDecode": LLaDA2UniImageDecode,
    "LLaDA2UniUnloadModel": LLaDA2UniUnloadModel,
}

NODE_DISPLAY_NAME_MAPPINGS = {
    "LLaDA2UniLoader": "LLaDA2.0_Uni Loader",
    "LLaDA2UniTextToImage": "LLaDA2.0_Uni Text-to-Image",
    "LLaDA2UniImageUnderstanding": "LLaDA2.0_Uni Image Understanding",
    "LLaDA2UniImageEditing": "LLaDA2.0_Uni Image Editing",
    "LLaDA2UniImageDecode": "LLaDA2.0_Uni Token Decoder",
    "LLaDA2UniUnloadModel": "LLaDA2.0_Uni Unload Model",
}

__all__ = ["NODE_CLASS_MAPPINGS", "NODE_DISPLAY_NAME_MAPPINGS"]