Hugging Face's logo

HY-2012
/
Sortformer.AXERA

Audio Classification
English
speaker-diarization
sortformer
axera
ax650
rttm
Model card Files
xet
 Community
Sortformer.AXERA
File size: 1,023 Bytes
7c268e9
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
{
  "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
}