audio-seg-diarization / config.json
nguyenvulebinh's picture
Update config.json
f2c53a2 verified
{
"_name_or_path": "audio_segmentation",
"architectures": [
"PyanNet"
],
"linear_hidden_size": 128,
"linear_numlayers": 2,
"lstm_batch_first": true,
"lstm_bidirectional": true,
"lstm_dropout": 0.5,
"lstm_hidden_size": 128,
"lstm_monolithic": true,
"lstm_num_layers": 4,
"max_spks": 3,
"model_type": "PyanNet",
"torch_dtype": "float32",
"transformers_version": "4.49.0"
}