File size: 1,605 Bytes
5a7ca3b | 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 58 59 60 61 62 | {
"architectures": [
"Nemotron3DiarizationForAudioFrameClassification"
],
"audio_config": {
"attention_dropout": 0.0,
"dtype": "float32",
"hidden_act": "gelu",
"hidden_size": 32,
"initializer_range": 0.02,
"intermediate_size": 64,
"max_position_embeddings": 512,
"model_type": "nemotron3_diarization_audio",
"num_attention_heads": 2,
"num_hidden_layers": 2,
"num_key_value_heads": 2,
"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": 4,
"chunk_right_context": 1,
"dtype": "float32",
"fifo_length": 4,
"head_config": {
"audio_hidden_size": 32,
"dtype": "float32",
"hidden_size": 16,
"model_type": "",
"num_speakers": 4,
"subsampling_factor": 8
},
"initializer_range": 0.02,
"model_type": "nemotron3_diarization",
"speaker_cache_update_period": 3,
"streaming_config": {
"dtype": "float32",
"fifo_length": 4,
"latest_frames_score_boost": 0.05,
"min_positive_scores_rate": 0.5,
"model_type": "",
"num_speakers": 4,
"prediction_score_threshold": 0.25,
"speaker_cache_length": 12,
"speaker_cache_silence_frames_per_speaker": 1,
"speaker_cache_update_period": 3,
"strong_boost_rate": 0.75,
"subsampling_factor": 8,
"weak_boost_rate": 1.5
},
"transformers.js_config": {
"dtype": "fp32",
"use_external_data_format": true
},
"transformers_version": "5.18.0.dev0"
}
|