16 lines
689 B
Python
16 lines
689 B
Python
from .hunyuan_image_3_gen import HunyuanImage3, HunyuanImage3ModelLoadingConfig # Import your actual node class
|
|
|
|
NODE_CLASS_MAPPINGS = {
|
|
"HunyuanImage3": HunyuanImage3, # The key is what will show up in ComfyUI interface
|
|
"HunyuanImage3ModelLoadingConfig": HunyuanImage3ModelLoadingConfig,
|
|
}
|
|
|
|
# Optional: If your node has display names different from class names
|
|
NODE_DISPLAY_NAME_MAPPINGS = {
|
|
"HunyuanImage3": "Hunyuan Image 3.0", # Optional, for better UI display
|
|
"HunyuanImage3ModelLoadingConfig": "Hunyuan Image 3.0 - model loading config",
|
|
}
|
|
|
|
# Required: To properly register your node
|
|
__all__ = ['NODE_CLASS_MAPPINGS', 'NODE_DISPLAY_NAME_MAPPINGS']
|