24 lines
846 B
Python
24 lines
846 B
Python
from .nodes import LoadDiTModel, LoadVAEModel, LoadTextEncoderModel, LoadGameImage, LoadMouseIcon, GameVideoGenerator, MatrixGameOutput
|
|
|
|
NODE_CLASS_MAPPINGS = {
|
|
"LoadDiTModel": LoadDiTModel,
|
|
"LoadVAEModel": LoadVAEModel,
|
|
"LoadTextEncoderModel": LoadTextEncoderModel,
|
|
"LoadGameImage": LoadGameImage,
|
|
"LoadMouseIcon": LoadMouseIcon,
|
|
"GameVideoGenerator": GameVideoGenerator,
|
|
"MatrixGameOutput": MatrixGameOutput,
|
|
}
|
|
|
|
NODE_DISPLAY_NAME_MAPPINGS = {
|
|
"LoadDiTModel": "Load DiT Model",
|
|
"LoadVAEModel": "Load VAE Model",
|
|
"LoadTextEncoderModel": "Load TextEncoder Model",
|
|
"LoadGameImage": "Load Game Image",
|
|
"LoadMouseIcon": "Load Mouse Icon",
|
|
"GameVideoGenerator": "Game Video Generator",
|
|
"MatrixGameOutput": "MatrixGame Output",
|
|
}
|
|
|
|
__all__ = ['NODE_CLASS_MAPPINGS', 'NODE_DISPLAY_NAME_MAPPINGS']
|