ComfyUI-FunAudioLLM_V2/__init__.py at main · zhilengjun/ComfyUI-FunAudioLLM_V2 · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
'''
Author: SpenserCai
Date: 2024-10-04 12:14:22
version:
LastEditors: SpenserCai
LastEditTime: 2024-10-04 22:29:33
Description: file content
'''
from .nodes.cosyvoice_nodes import *
from .nodes.sensevoice_nodes import *
from .nodes.inspiremusic_nodes import *

NODE_CONFIG = {
    "CosyVoice2ZeroShotNode": {
        "class": CosyVoice2ZeroShotNode,
        "name": "CosyVoice2 音色克隆"
    },
    "CosyVoice2CrossLingualNode": {
        "class": CosyVoice2CrossLingualNode,
        "name": "CosyVoice2 跨语言克隆"
    },
    "CosyVoice2InstructNode": {
        "class": CosyVoice2InstructNode,
        "name": "CosyVoice2 自然语言控制"
    },
    "CosyVoiceZeroShotNode": {
        "class": CosyVoiceZeroShotNode,
        "name": "CosyVoice 音色克隆"
    },
    "CosyVoiceSFTNode": {
        "class": CosyVoiceSFTNode,
        "name": "CosyVoice 预训练音色"
    },
    "CosyVoiceCrossLingualNode": {
        "class": CosyVoiceCrossLingualNode,
        "name": "CosyVoice 跨语言克隆"
    },
    "CosyVoiceInstructNode": {
        "class": CosyVoiceInstructNode,
        "name": "CosyVoice 自然语言控制"
    },
    "CosyVoiceSaveSpeakerModelNode": {
        "class": CosyVoiceSaveSpeakerModelNode,
        "name": "CosyVoice 保存说话人模型"
    },
    "CosyVoiceLoadSpeakerModelNode": {
        "class": CosyVoiceLoadSpeakerModelNode,
        "name": "CosyVoice 加载说话人模型"
    },
    "CosyVoiceLoadSpeakerModelFromUrlNode": {
        "class": CosyVoiceLoadSpeakerModelFromUrlNode,
        "name": "CosyVoice 从URL加载说话人模型"
    },
    "SenseVoiceNode": {
        "class": SenseVoiceNode,
        "name": "SenseVoice 语音识别"
    },

    "TextToMusic": {
        "class": TextToMusic,
        "name": "文本生成音乐"
    },

    "MusicContinuation": {
        "class": MusicContinuation,
        "name": "音乐延续"
    },
}

def generate_node_mappings(node_config):
    node_class_mappings = {}
    node_display_name_mappings = {}

    for node_name, node_info in node_config.items():
        node_class_mappings[node_name] = node_info["class"]
        node_display_name_mappings[node_name] = node_info.get("name", node_info["class"].__name__)

    return node_class_mappings, node_display_name_mappings

NODE_CLASS_MAPPINGS, NODE_DISPLAY_NAME_MAPPINGS = generate_node_mappings(NODE_CONFIG)

WEB_DIRECTORY = "./web"

__all__ = ["NODE_CLASS_MAPPINGS", "NODE_DISPLAY_NAME_MAPPINGS", "WEB_DIRECTORY"]