parley-models / gpu16 /diarization /diarization.json
killameep's picture
Add gpu16 variant (fp16, LayerNorms in fp32) for DirectML GPUs
7bd332f verified
Raw History Blame Contribute Delete
563 Bytes
{
"num_speakers": 8,
"subsampling_factor": 8,
"speaker_cache_length": 264,
"fifo_length": 264,
"speaker_cache_update_period": 222,
"silence_frames_per_speaker": 1,
"prediction_score_threshold": 0.25,
"latest_frames_score_boost": 0.05,
"min_positive_scores_rate": 0.5,
"strong_boost_rate": 0.75,
"weak_boost_rate": 1.5,
"hidden_size": 512,
"streaming_modes": {
"low_latency": [
9,
4
],
"very_low_latency": [
6,
2
],
"ultra_low_latency": [
3,
1
]
}
}