20 lines
830 B
Python
20 lines
830 B
Python
from .nodes.volcengine_seedream_v3 import VolcengineSeeDreamV3Node
|
|
from .nodes.volcengine_i2v_s2pro import VolcengineI2VS2Pro
|
|
from .nodes.volcengine_img_edit_v3 import VolcengineImgEditV3
|
|
from .nodes.volcengine_doubao_seedance import VolcengineDoubaoSeedance
|
|
|
|
NODE_CLASS_MAPPINGS = {
|
|
"volcengine-seedream-v3": VolcengineSeeDreamV3Node,
|
|
"volcengine-i2v-s2pro": VolcengineI2VS2Pro,
|
|
"volcengine-img-edit-v3": VolcengineImgEditV3,
|
|
"volcengine-doubao-seedance": VolcengineDoubaoSeedance
|
|
}
|
|
|
|
NODE_DISPLAY_NAME_MAPPINGS = {
|
|
"volcengine-seedream-v3": "Volcengine SeeDream V3",
|
|
"volcengine-i2v-s2pro": "Volcengine I2V S2.0Pro",
|
|
"volcengine-img-edit-v3": "Volcengine Img Edit V3.0",
|
|
"volcengine-doubao-seedance": "Volcengine Doubao Seedance"
|
|
}
|
|
|
|
__all__ = ['NODE_CLASS_MAPPINGS', 'NODE_DISPLAY_NAME_MAPPINGS'] |