File size: 1,673 Bytes
2b9fdf9
 
a4308eb
2b9fdf9
 
 
a4308eb
2b9fdf9
 
 
 
 
 
 
 
 
 
 
 
 
a4308eb
2b9fdf9
 
 
 
 
 
 
 
 
 
 
 
 
 
a4308eb
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
2b9fdf9
 
 
 
 
 
 
 
 
 
 
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
{
  "format": "onnx-magpie-export-summary-v1",
  "nemo_path": "C:\\Users\\maxw6\\Projects\\onnx-magpie-artifacts\\magpie_tts_multilingual_357m.nemo",
  "opset": 18,
  "graphs": [
    {
      "path": "magpie_text_context.onnx",
      "inputs": [
        "text",
        "text_lens",
        "speaker_indices"
      ],
      "outputs": [
        "cond",
        "cond_mask",
        "additional_decoder_input",
        "additional_decoder_mask"
      ]
    },
    {
      "path": "magpie_decoder_prefix.onnx",
      "inputs": [
        "audio_tokens",
        "audio_tokens_lens",
        "cond",
        "cond_mask",
        "additional_decoder_input",
        "additional_decoder_mask"
      ],
      "outputs": [
        "next_logits",
        "decoder_state"
      ]
    },
    {
      "path": "magpie_decoder_kv_init.onnx",
      "inputs": [
        "cond",
        "cond_mask",
        "additional_decoder_input",
        "additional_decoder_mask"
      ],
      "outputs": [
        "present_cache",
        "present_value_cache"
      ]
    },
    {
      "path": "magpie_decoder_kv_step.onnx",
      "inputs": [
        "audio_tokens",
        "audio_tokens_lens",
        "cache_position",
        "cond",
        "cond_mask",
        "past_cache",
        "past_value_cache"
      ],
      "outputs": [
        "next_logits",
        "present_cache",
        "present_value_cache"
      ]
    },
    {
      "path": "magpie_codec_decoder.onnx",
      "inputs": [
        "codes",
        "codes_lens"
      ],
      "outputs": [
        "audio",
        "audio_lens"
      ]
    }
  ]
}