nodes/Yuan-ManX-ComfyUI-Matrix-Game
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Update __init__.py

Browse Source
This commit is contained in:
Yuan-Man
2025-05-13 16:04:59 +08:00
committed by GitHub
parent 316c441dc9
commit 1a57207bcb
1 changed files with 15 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
__init__.py
+15 -3
View File
@@ -1,11 +1,23 @@
from .nodes import PlayGame
from .nodes import LoadDiTModel, LoadVAEModel, LoadTextEncoderModel, LoadGameImage, LoadMouseIcon, GameVideoGenerator, MatrixGameOutput
NODE_CLASS_MAPPINGS = {
"PlayGame": PlayGame,
"LoadDiTModel": LoadDiTModel,
"LoadVAEModel": LoadVAEModel,
"LoadTextEncoderModel": LoadTextEncoderModel,
"LoadGameImage": LoadGameImage,
"LoadMouseIcon": LoadMouseIcon,
"GameVideoGenerator": GameVideoGenerator,
"MatrixGameOutput": MatrixGameOutput,
}
NODE_DISPLAY_NAME_MAPPINGS = {
"PlayGame": "Play Game",
"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']