saute / config.json
JustinDuc's picture
Update config.json
91a2e1f verified
{
"architectures": [
"UtteranceEmbedings"
],
"attention_probs_dropout_prob": 0.1,
"auto_map": {
"AutoConfig": "saute_config.SAUTEConfig",
"AutoModel": "saute_model.UtteranceEmbedings"
},
"hidden_dropout_prob": 0.1,
"hidden_size": 768,
"intermediate_size": 3072,
"max_edu_length": 128,
"max_edus_per_dialog": 100,
"max_position_embeddings": 512,
"max_speakers": 200,
"model_type": "saute",
"num_attention_heads": 8,
"num_edu_layers": 2,
"num_hidden_layers": 3,
"num_speaker_embeddings": 512,
"num_token_layers": 2,
"speaker_embeddings_size": 768,
"torch_dtype": "float32",
"transformers_version": "4.52.4",
"vocab_size": 30522
}