models/model_meta.json

{
  "model": "nvidia/diar_streaming_sortformer_4spk-v2.1",
  "contract": "split-preencode-encoder",
  "simplified": true,
  "config": {
    "chunk_len": 6,
    "chunk_left_context": 1,
    "chunk_right_context": 7,
    "fifo_len": 188,
    "spkcache_len": 188,
    "spkcache_update_period": 144
  },
  "preencode": {
    "file": "preencode.onnx",
    "inputs": {
      "chunk": [
        1,
        112,
        128
      ]
    },
    "outputs": {
      "chunk_pre_encode_embs": [
        1,
        14,
        512
      ]
    },
    "parity_min_cosine": 0.9999999403953552
  },
  "encoder": {
    "file": "encoder.onnx",
    "inputs": {
      "seq": [
        1,
        390,
        512
      ],
      "total_lengths": [
        1
      ]
    },
    "outputs": {
      "spkcache_fifo_chunk_preds": [
        1,
        390,
        4
      ]
    },
    "parity_min_cosine": 0.9999998807907104
  },
  "state_len": 376,
  "chunk_mel_frames": 112,
  "frame_duration_s": 0.08,
  "sample_rate": 16000,
  "mel_features": 128
}