Files

39 lines
1.8 KiB
Python

from .nodes.image_to_3d import ImageTo3DNode
from .nodes.load_image import LoadImage
from .nodes.image_pixelator import ImagePixelatorNode
from .nodes.load_multiple_images import LoadMultipleImages
from .nodes.image_slic_pixelator import SLICPixelatorNode
from .nodes.seed_save_audio import SeedSaveAudio
from .nodes.image_transparency_crop import ImageTransparencyCropNode
from .nodes.qwen_vl_api import QwenVLAPINode
from .nodes.siliconflow_vl_api import SiliconFlowVLAPINode
from .nodes.screen_mode_remove_black import ScreenModeRemoveBlackNode
# 映射节点类到其在 ComfyUI 中的显示名称
NODE_CLASS_MAPPINGS = {
"Seed-Nodes: LoadImage": LoadImage,
"Seed-Nodes: ImagePixelator": ImagePixelatorNode,
"Seed-Nodes: LoadMultipleImages": LoadMultipleImages,
"Seed-Nodes: SLICPixelator": SLICPixelatorNode,
"Seed-Nodes: ImageTo3D": ImageTo3DNode,
"Seed-Nodes: SeedSaveAudio": SeedSaveAudio,
"Seed-Nodes: ImageTransparencyCrop": ImageTransparencyCropNode,
"Seed-Nodes: QwenVLAPI": QwenVLAPINode,
"Seed-Nodes: SiliconFlowVLAPI": SiliconFlowVLAPINode,
"Seed-Nodes: ScreenModeRemoveBlack": ScreenModeRemoveBlackNode,
}
# 定义节点在 ComfyUI 中的显示名称
NODE_DISPLAY_NAME_MAPPINGS = {
"Seed-Nodes: LoadImage": "Seed-Nodes: LoadImage",
"Seed-Nodes: ImagePixelator": "Seed-Nodes: ImagePixelator",
"Seed-Nodes: LoadMultipleImages": "Seed-Nodes: LoadMultipleImages",
"Seed-Nodes: SLICPixelator": "Seed-Nodes: SLICPixelator",
"Seed-Nodes: ImageTo3D": "Seed-Nodes: ImageTo3D",
"Seed-Nodes: SeedSaveAudio": "Seed-Nodes: SeedSaveAudio",
"Seed-Nodes: ImageTransparencyCrop": "Seed-Nodes: ImageTransparencyCrop",
"Seed-Nodes: QwenVLAPI": "Seed-Nodes: QwenVLAPI",
"Seed-Nodes: SiliconFlowVLAPI": "Seed-Nodes: SiliconFlowVLAPI",
"Seed-Nodes: ScreenModeRemoveBlack": "Seed-Nodes: Screen模式去黑底",
}