config.json

{
  "source_checkpoint": {
    "repository": "nvidia/Nemotron-3-Diarization",
    "revision": "98fcee8e866bb534eee85d1eb3817170700ae717",
    "sha256": "9930e1719c85b35da083c557cdbd6503072c311ffe8bc9cbd3371d2609e21d0f"
  },
  "precision": {
    "weights_storage": "fp16",
    "compute": "fp32"
  },
  "minimum_deployment_target": "macOS14 / iOS17",
  "streaming_profile": {
    "name": "offline",
    "chunk_len": 340,
    "chunk_left_context": 0,
    "chunk_right_context": 40,
    "fifo_len": 40,
    "spkcache_len": 264,
    "spkcache_update_period": 300,
    "frame_seconds": 0.08,
    "subsampling_factor": 8
  },
  "speaker_cache": {
    "n_spk": 8,
    "spkcache_sil_frames_per_spk": 1,
    "use_learnable_sil_emb": true,
    "learnable_sil_emb_file": "learnable_sil_emb.f32",
    "learnable_sil_emb_dim": 512,
    "learnable_sil_emb_format": "float32 little-endian, raw"
  },
  "postprocessing": {
    "pred_score_threshold": 0.25,
    "onset": 0.5,
    "offset": 0.5,
    "pad_onset": 0.0,
    "pad_offset": 0.0,
    "min_duration_on": 0.0,
    "min_duration_off": 0.0
  },
  "inputs": {
    "chunk": [
      1,
      3040,
      128
    ],
    "chunk_lengths": [
      1
    ],
    "spkcache": [
      1,
      264,
      512
    ],
    "spkcache_lengths": [
      1
    ],
    "fifo": [
      1,
      40,
      512
    ],
    "fifo_lengths": [
      1
    ]
  },
  "outputs": {
    "spkcache_fifo_chunk_preds": [
      1,
      684,
      8
    ],
    "chunk_pre_encode_embs": [
      1,
      380,
      512
    ],
    "chunk_pre_encode_lengths": [
      1
    ]
  },
  "sha256": {
    "learnable_sil_emb.f32": "d4417b3c0eabdf7c47032fac2b5b5a7ee83d819a6ddda8fd8eaf74e2b5cc4ac7",
    "nemotron_3_diarization.mlpackage/Data/com.apple.CoreML/model.mlmodel": "3da7a6f70a632e9079116f9c14bf7afbf5ff69ac5b28e90dd4277a9ce41f8642",
    "nemotron_3_diarization.mlpackage/Data/com.apple.CoreML/weights/weight.bin": "b7c50bd483210f206014548c0cc2fc7b3601de012ab86b2e49f8a911db9a7d40",
    "nemotron_3_diarization.mlpackage/Manifest.json": "87c6d992c3300934d6b0205dbcf0a34df93e4b5716e6ccd1de846147b2937705"
  }
}