magpie.export.json

{
  "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"
      ]
    }
  ]
}