{
  "format_version": 1,
  "layers": 8,
  "kv_heads": 16,
  "head_dim": 64,
  "window": 72,
  "states": [
    {
      "name": "pre_conv",
      "shape": [
        1,
        512,
        2
      ]
    },
    {
      "name": "upsample.0.1.dwconv",
      "shape": [
        1,
        1024,
        6
      ]
    },
    {
      "name": "upsample.1.1.dwconv",
      "shape": [
        1,
        1024,
        6
      ]
    },
    {
      "name": "decoder.0",
      "shape": [
        1,
        1024,
        6
      ]
    },
    {
      "name": "decoder.1.block.1",
      "shape": [
        1,
        768,
        8
      ]
    },
    {
      "name": "decoder.1.block.2.conv1",
      "shape": [
        1,
        768,
        6
      ]
    },
    {
      "name": "decoder.1.block.3.conv1",
      "shape": [
        1,
        768,
        18
      ]
    },
    {
      "name": "decoder.1.block.4.conv1",
      "shape": [
        1,
        768,
        54
      ]
    },
    {
      "name": "decoder.2.block.1",
      "shape": [
        1,
        384,
        5
      ]
    },
    {
      "name": "decoder.2.block.2.conv1",
      "shape": [
        1,
        384,
        6
      ]
    },
    {
      "name": "decoder.2.block.3.conv1",
      "shape": [
        1,
        384,
        18
      ]
    },
    {
      "name": "decoder.2.block.4.conv1",
      "shape": [
        1,
        384,
        54
      ]
    },
    {
      "name": "decoder.3.block.1",
      "shape": [
        1,
        192,
        4
      ]
    },
    {
      "name": "decoder.3.block.2.conv1",
      "shape": [
        1,
        192,
        6
      ]
    },
    {
      "name": "decoder.3.block.3.conv1",
      "shape": [
        1,
        192,
        18
      ]
    },
    {
      "name": "decoder.3.block.4.conv1",
      "shape": [
        1,
        192,
        54
      ]
    },
    {
      "name": "decoder.4.block.1",
      "shape": [
        1,
        96,
        3
      ]
    },
    {
      "name": "decoder.4.block.2.conv1",
      "shape": [
        1,
        96,
        6
      ]
    },
    {
      "name": "decoder.4.block.3.conv1",
      "shape": [
        1,
        96,
        18
      ]
    },
    {
      "name": "decoder.4.block.4.conv1",
      "shape": [
        1,
        96,
        54
      ]
    },
    {
      "name": "decoder.6",
      "shape": [
        1,
        96,
        6
      ]
    }
  ],
  "inputs": [
    "codes",
    "offset",
    "past",
    "conv_0",
    "conv_1",
    "conv_2",
    "conv_3",
    "conv_4",
    "conv_5",
    "conv_6",
    "conv_7",
    "conv_8",
    "conv_9",
    "conv_10",
    "conv_11",
    "conv_12",
    "conv_13",
    "conv_14",
    "conv_15",
    "conv_16",
    "conv_17",
    "conv_18",
    "conv_19",
    "conv_20"
  ],
  "outputs": [
    "audio",
    "present",
    "next_conv_0",
    "next_conv_1",
    "next_conv_2",
    "next_conv_3",
    "next_conv_4",
    "next_conv_5",
    "next_conv_6",
    "next_conv_7",
    "next_conv_8",
    "next_conv_9",
    "next_conv_10",
    "next_conv_11",
    "next_conv_12",
    "next_conv_13",
    "next_conv_14",
    "next_conv_15",
    "next_conv_16",
    "next_conv_17",
    "next_conv_18",
    "next_conv_19",
    "next_conv_20"
  ]
}