Files
Aryan185-ComfyUI-VertexAPI/__init__.py
T
Aryan185 f439b28f6b - Streamlined code.
- Updated pyproject.toml and requirements.txt.
2026-01-22 13:09:47 +00:00

30 lines
719 B
Python

from . import (
gemini_node_vertex,
gemini_segment_vertex,
nano_banana_vertex,
gemini_tts_vertex,
veo_vertex,
gemini_diarisation_vertex,
imagen_edit_vertex,
imagen_vertex,
)
NODE_CLASS_MAPPINGS = {}
NODE_DISPLAY_NAME_MAPPINGS = {}
modules = [
gemini_node_vertex,
gemini_segment_vertex,
nano_banana_vertex,
gemini_tts_vertex,
veo_vertex,
gemini_diarisation_vertex,
imagen_edit_vertex,
imagen_vertex,
]
for module in modules:
NODE_CLASS_MAPPINGS.update(getattr(module, "NODE_CLASS_MAPPINGS", {}))
NODE_DISPLAY_NAME_MAPPINGS.update(getattr(module, "NODE_DISPLAY_NAME_MAPPINGS", {}))
__all__ = ["NODE_CLASS_MAPPINGS", "NODE_DISPLAY_NAME_MAPPINGS"]