magpie_decoder_prefix.manifest.json

{
  "model_path": "models\\magpie_decoder_prefix.onnx",
  "ir_version": 8,
  "producer_name": "pytorch",
  "producer_version": "2.10.0",
  "opsets": [
    {
      "domain": "ai.onnx",
      "version": 18
    }
  ],
  "inputs": [
    {
      "name": "audio_tokens",
      "elem_type": "INT64",
      "shape": [
        1,
        8,
        "audio_time"
      ]
    },
    {
      "name": "audio_tokens_lens",
      "elem_type": "INT64",
      "shape": [
        1
      ]
    },
    {
      "name": "cond",
      "elem_type": "FLOAT",
      "shape": [
        1,
        "text_time",
        768
      ]
    },
    {
      "name": "cond_mask",
      "elem_type": "BOOL",
      "shape": [
        1,
        "text_time"
      ]
    },
    {
      "name": "additional_decoder_input",
      "elem_type": "FLOAT",
      "shape": [
        1,
        110,
        768
      ]
    },
    {
      "name": "additional_decoder_mask",
      "elem_type": "BOOL",
      "shape": [
        1,
        110
      ]
    }
  ],
  "outputs": [
    {
      "name": "next_logits",
      "elem_type": "FLOAT",
      "shape": [
        "Gathernext_logits_dim_0",
        16192
      ]
    },
    {
      "name": "decoder_state",
      "elem_type": "FLOAT",
      "shape": [
        "Gathernext_logits_dim_0",
        768
      ]
    }
  ]
}