hf-transformers-bot's picture
Update tiny models for GraniteSpeech5ForCTC
2f59f42 verified
Raw History Blame Contribute Delete
944 Bytes
{
"architectures": [
"GraniteSpeech5ForCTC"
],
"ctc_loss_reduction": "mean",
"ctc_zero_infinity": true,
"dtype": "float32",
"encoder_config": {
"activation_dropout": 0.0,
"attention_bias": true,
"attention_dropout": 0.0,
"context_size": 13,
"conv_expansion_factor": 2,
"conv_kernel_size": 7,
"head_dim": 16,
"hidden_act": "silu",
"hidden_size": 32,
"initializer_range": 0.02,
"intermediate_size": 64,
"max_position_embeddings": 64,
"model_type": "granite_speech5_encoder",
"num_attention_heads": 2,
"num_hidden_layers": 2,
"num_key_value_heads": 2,
"num_mel_bins": 6,
"subsample_layers": [
0
],
"vocab_size": 16384
},
"initializer_range": 0.02,
"model_type": "granite_speech5_ctc",
"pad_token_id": 0,
"tie_word_embeddings": true,
"transformers_version": "5.19.0.dev0",
"vocab_size": 16384
}