magpie_decoder_kv_init.manifest.json

{
  "model_path": "magpie_decoder_kv_init.onnx",
  "ir_version": 8,
  "producer_name": "pytorch",
  "producer_version": "2.10.0",
  "opsets": [
    {
      "domain": "ai.onnx",
      "version": 18
    }
  ],
  "inputs": [
    {
      "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": "present_cache",
      "elem_type": "FLOAT",
      "shape": [
        12,
        1,
        192,
        12,
        64
      ]
    },
    {
      "name": "present_value_cache",
      "elem_type": "FLOAT",
      "shape": [
        12,
        1,
        192,
        12,
        64
      ]
    }
  ]
}