{
  "hidden_size": 896,
  "layers": 24,
  "attention_heads": 14,
  "kv_heads": 2,
  "head_dimension": 64,
  "rope_theta": 1000000,
  "rms_norm_epsilon": 1e-06,
  "codec_cache_capacity": 64,
  "codec_frame_capacity": 32,
  "codec_attention_window": 64,
  "codec_fft_size": 3528,
  "codec_hop_length": 882,
  "codec_padding": 1323,
  "codec_upsample_factor": 4,
  "codec_upsample_align_corners": false,
  "codec_group_layers": 12,
  "precision": "w8a16",
  "normalization": "scaled-fp16",
  "attention": "repeated-kv",
  "projection_reduction_channels": 256,
  "preferred_compute_unit_kind": "ane",
  "schema_version": 1,
  "model_family": "mingTinyTTSAudio",
  "sample_rate": 44100,
  "channels": 1,
  "latent_dimension": 64,
  "condition_dimension": 1536,
  "flow_hidden_size": 1024,
  "flow_layers": 16,
  "aggregator_hidden_size": 1024,
  "aggregator_layers": 8,
  "flow_projection_reduction_channels": null,
  "resources": "resources.json",
  "resource_data": "resources.f32",
  "codec_window_resource": "audio.decoder.head.istft.window",
  "stop_projection": "stop_head",
  "speaker_projection": "spk_head",
  "time_features_resource": "flow.time_features",
  "timesteps_resource": "flow.timesteps",
  "assets": [
    {
      "name": "aggregator",
      "functions": [
        {
          "name": "main",
          "inputs": {
            "latent": [
              1,
              64,
              1,
              1
            ]
          },
          "outputs": [
            "hidden"
          ],
          "output_shapes": {
            "hidden": [
              1,
              1536,
              1,
              1
            ]
          }
        }
      ]
    },
    {
      "name": "flow",
      "functions": [
        {
          "name": "main",
          "inputs": {
            "noise": [
              1,
              64,
              1,
              1
            ],
            "condition": [
              1,
              1536,
              1,
              1
            ],
            "history": [
              1,
              64,
              1,
              1
            ],
            "time_features": [
              1,
              1024,
              1,
              1
            ]
          },
          "outputs": [
            "velocity"
          ],
          "output_shapes": {
            "velocity": [
              2,
              64,
              1,
              1
            ]
          }
        }
      ]
    },
    {
      "name": "codec-0",
      "functions": [
        {
          "name": "main",
          "inputs": {
            "hidden": [
              1,
              896,
              1,
              32
            ],
            "cos": [
              1,
              64,
              1,
              32
            ],
            "sin": [
              1,
              64,
              1,
              32
            ],
            "mask": [
              1,
              96,
              1,
              32
            ],
            "keys": [
              1,
              1536,
              1,
              64
            ],
            "values": [
              1,
              1536,
              1,
              64
            ]
          },
          "outputs": [
            "hidden",
            "next_keys",
            "next_values"
          ],
          "output_shapes": {
            "hidden": [
              1,
              896,
              1,
              32
            ],
            "next_keys": [
              1,
              1536,
              1,
              32
            ],
            "next_values": [
              1,
              1536,
              1,
              32
            ]
          }
        }
      ],
      "family": "codec",
      "layer_start": 0,
      "layers": 12,
      "cache_capacity": 64,
      "window": 64
    },
    {
      "name": "codec-1",
      "functions": [
        {
          "name": "main",
          "inputs": {
            "hidden": [
              1,
              896,
              1,
              32
            ],
            "cos": [
              1,
              64,
              1,
              32
            ],
            "sin": [
              1,
              64,
              1,
              32
            ],
            "mask": [
              1,
              96,
              1,
              32
            ],
            "keys": [
              1,
              1536,
              1,
              64
            ],
            "values": [
              1,
              1536,
              1,
              64
            ]
          },
          "outputs": [
            "hidden",
            "next_keys",
            "next_values",
            "spectrum"
          ],
          "output_shapes": {
            "hidden": [
              1,
              896,
              1,
              32
            ],
            "next_keys": [
              1,
              1536,
              1,
              32
            ],
            "next_values": [
              1,
              1536,
              1,
              32
            ],
            "spectrum": [
              1,
              3530,
              1,
              32
            ]
          }
        }
      ],
      "family": "codec",
      "layer_start": 12,
      "layers": 12,
      "cache_capacity": 64,
      "window": 64
    },
    {
      "name": "encoder-0",
      "functions": [
        {
          "name": "main",
          "inputs": {
            "hidden": [
              1,
              896,
              1,
              32
            ],
            "cos": [
              1,
              64,
              1,
              32
            ],
            "sin": [
              1,
              64,
              1,
              32
            ],
            "mask": [
              1,
              96,
              1,
              32
            ],
            "keys": [
              1,
              1536,
              1,
              64
            ],
            "values": [
              1,
              1536,
              1,
              64
            ]
          },
          "outputs": [
            "hidden",
            "next_keys",
            "next_values"
          ],
          "output_shapes": {
            "hidden": [
              1,
              896,
              1,
              32
            ],
            "next_keys": [
              1,
              1536,
              1,
              32
            ],
            "next_values": [
              1,
              1536,
              1,
              32
            ]
          }
        }
      ],
      "family": "encoder",
      "layer_start": 0,
      "layers": 12,
      "cache_capacity": 64,
      "window": 64
    },
    {
      "name": "encoder-1",
      "functions": [
        {
          "name": "main",
          "inputs": {
            "hidden": [
              1,
              896,
              1,
              32
            ],
            "cos": [
              1,
              64,
              1,
              32
            ],
            "sin": [
              1,
              64,
              1,
              32
            ],
            "mask": [
              1,
              96,
              1,
              32
            ],
            "keys": [
              1,
              1536,
              1,
              64
            ],
            "values": [
              1,
              1536,
              1,
              64
            ]
          },
          "outputs": [
            "hidden",
            "next_keys",
            "next_values"
          ],
          "output_shapes": {
            "hidden": [
              1,
              896,
              1,
              32
            ],
            "next_keys": [
              1,
              1536,
              1,
              32
            ],
            "next_values": [
              1,
              1536,
              1,
              32
            ]
          }
        }
      ],
      "family": "encoder",
      "layer_start": 12,
      "layers": 12,
      "cache_capacity": 64,
      "window": 64
    },
    {
      "name": "encoder-aggregator-0",
      "functions": [
        {
          "name": "main",
          "inputs": {
            "hidden": [
              1,
              896,
              1,
              40
            ],
            "cos": [
              1,
              64,
              1,
              40
            ],
            "sin": [
              1,
              64,
              1,
              40
            ],
            "mask": [
              1,
              104,
              1,
              40
            ],
            "keys": [
              1,
              512,
              1,
              64
            ],
            "values": [
              1,
              512,
              1,
              64
            ]
          },
          "outputs": [
            "hidden",
            "next_keys",
            "next_values"
          ],
          "output_shapes": {
            "hidden": [
              1,
              896,
              1,
              40
            ],
            "next_keys": [
              1,
              512,
              1,
              40
            ],
            "next_values": [
              1,
              512,
              1,
              40
            ]
          }
        }
      ],
      "family": "encoder-aggregator",
      "layer_start": 0,
      "layers": 4,
      "cache_capacity": 64,
      "window": 64
    }
  ]
}