pixelworld AI
797d0e08f5
- Add byteplus_chat_node to node_list for automatic loading - Enables ComfyUI to discover and load the new Seed 1.6 Chat node
25 lines
623 B
Python
25 lines
623 B
Python
import importlib
|
|
import importlib.util
|
|
|
|
node_list = [
|
|
"byteplus_video_node",
|
|
"video_to_frames_node",
|
|
"byteplus_image_node",
|
|
"byteplus_chat_node",
|
|
]
|
|
|
|
NODE_CLASS_MAPPINGS = {}
|
|
NODE_DISPLAY_NAME_MAPPINGS = {}
|
|
|
|
for module_name in node_list:
|
|
imported_module = importlib.import_module(f".nodes.{module_name}", __name__)
|
|
|
|
NODE_CLASS_MAPPINGS = {**NODE_CLASS_MAPPINGS, **imported_module.NODE_CLASS_MAPPINGS}
|
|
NODE_DISPLAY_NAME_MAPPINGS = {
|
|
**NODE_DISPLAY_NAME_MAPPINGS,
|
|
**imported_module.NODE_DISPLAY_NAME_MAPPINGS,
|
|
}
|
|
|
|
|
|
__all__ = ["NODE_CLASS_MAPPINGS", "NODE_DISPLAY_NAME_MAPPINGS"]
|