File size: 1,458 Bytes
aaba974 | 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 | {
"architectures": [
"Nemotron3DiarizationForAudioFrameClassification"
],
"audio_config": {
"attention_dropout": 0.0,
"hidden_act": "gelu",
"hidden_size": 512,
"initializer_range": 0.02,
"intermediate_size": 2048,
"max_position_embeddings": 5000,
"model_type": "nemotron3_diarization_audio",
"num_attention_heads": 8,
"num_hidden_layers": 31,
"num_key_value_heads": 8,
"num_mel_bins": 128,
"partial_rotary_factor": 1.0,
"rope_parameters": {
"partial_rotary_factor": 1.0,
"rope_theta": 10000.0,
"rope_type": "default"
},
"subsampling_factor": 8
},
"chunk_length": 340,
"chunk_right_context": 40,
"dtype": "float32",
"fifo_length": 40,
"head_config": {
"audio_hidden_size": 512,
"hidden_size": 192,
"model_type": "",
"num_speakers": 8,
"subsampling_factor": 8
},
"initializer_range": 0.02,
"model_type": "nemotron3_diarization",
"speaker_cache_update_period": 300,
"streaming_config": {
"fifo_length": 264,
"latest_frames_score_boost": 0.05,
"min_positive_scores_rate": 0.5,
"model_type": "",
"num_speakers": 8,
"prediction_score_threshold": 0.25,
"speaker_cache_length": 264,
"speaker_cache_silence_frames_per_speaker": 1,
"speaker_cache_update_period": 222,
"strong_boost_rate": 0.75,
"subsampling_factor": 8,
"weak_boost_rate": 1.5
},
"transformers_version": "5.18.0.dev0"
}
|