32 lines
1.0 KiB
Python
32 lines
1.0 KiB
Python
from .nodes import (
|
|
LongCatCheckpointLoader,
|
|
TextEncodeLongCatImage,
|
|
TextEncodeLongCatImageEdit,
|
|
VAEEncodeLongCat,
|
|
VAEDecodeLongCat,
|
|
LongCatSampler,
|
|
LongCatImageSizeScale,
|
|
)
|
|
|
|
NODE_CLASS_MAPPINGS = {
|
|
"LongCatCheckpointLoader": LongCatCheckpointLoader,
|
|
"TextEncodeLongCatImage": TextEncodeLongCatImage,
|
|
"TextEncodeLongCatImageEdit": TextEncodeLongCatImageEdit,
|
|
"VAEEncodeLongCat": VAEEncodeLongCat,
|
|
"VAEDecodeLongCat": VAEDecodeLongCat,
|
|
"LongCatSampler": LongCatSampler,
|
|
"LongCatImageSizeScale": LongCatImageSizeScale,
|
|
}
|
|
|
|
NODE_DISPLAY_NAME_MAPPINGS = {
|
|
"LongCatCheckpointLoader": "LongCat Checkpoint Loader",
|
|
"TextEncodeLongCatImage": "LongCat Text Encode (T2I)",
|
|
"TextEncodeLongCatImageEdit": "LongCat Text Encode (Edit)",
|
|
"VAEEncodeLongCat": "LongCat VAE Encode",
|
|
"VAEDecodeLongCat": "LongCat VAE Decode",
|
|
"LongCatSampler": "LongCat Sampler",
|
|
"LongCatImageSizeScale": "LongCat Image Size Scale",
|
|
}
|
|
|
|
__all__ = ["NODE_CLASS_MAPPINGS", "NODE_DISPLAY_NAME_MAPPINGS"]
|