| { |
| "model_type": "mateli-rnnt", |
| "sample_rate": 16000, |
| "features": { |
| "type": "log-mel", |
| "n_mels": 80, |
| "n_fft": 512, |
| "win_length": 400, |
| "hop_length": 160, |
| "window": "periodic hann", |
| "center": true, |
| "pad_mode": "reflect", |
| "log_eps": 1e-06, |
| "filterbank": "mel_fb.f32" |
| }, |
| "chunk": { |
| "mel_frames": 32, |
| "encoder_frames": 8, |
| "frame_seconds": 0.04 |
| }, |
| "encoder": { |
| "files": {"int8": "encoder.int8.onnx", "fp32": "encoder.onnx"}, |
| "inputs": ["mel", "n_valid", "gru_h", "conv_cache", "attn_kv", "attn_len", "mel_cache", "sub_cache"], |
| "outputs": ["enc", "enc_proj", "gru_h_out", "conv_cache_out", "attn_kv_out", "attn_len_out", "mel_cache_out", "sub_cache_out"], |
| "state_shapes": { |
| "gru_h": [16, 1, 384], |
| "conv_cache": [16, 384, 14], |
| "attn_kv": [4, 2, 64, 384], |
| "attn_len": [1], |
| "mel_cache": [1, 1, 80], |
| "sub_cache": [1, 384, 1, 40] |
| } |
| }, |
| "pred_step": {"file": "pred_step.onnx", "inputs": ["token", "h"], "outputs": ["pred_proj", "h_out"], "hidden": 384}, |
| "joint": {"file": "joint.onnx", "inputs": ["enc_proj", "pred_proj"], "outputs": ["logits"]}, |
| "vocab": ["<blank>", " ", "a", "b", "c", "d", "e", "f", "g", "h", "i", "j", "k", "l", "m", "n", "o", "p", "q", "r", "s", "t", "u", "v", "w", "x", "y", "z", "å", "ä", "ö", "0", "1", "2", "3", "4", "5", "6", "7", "8", "9", ".", "?", "!"], |
| "blank_id": 0, |
| "space_id": 1, |
| "max_symbols_per_frame": 8 |
| } |
|
|
