moonshine_tiny / kf_config.json
IMvision12's picture
Upgrade kf_config.json to v2 (nested config + weight_dtype)
3f346a8 verified
Raw
History Blame Contribute Delete
946 Bytes
{
"library_name": "kerasformers",
"kerasformers_version": "1.2.1",
"model_module": "kerasformers.models.moonshine",
"model_class": "MoonshineModel",
"variant": "moonshine_tiny",
"weights": "model.weights.h5",
"schema_version": 2,
"weight_dtype": "float32",
"model_type": "moonshine",
"audio_config": {
"num_layers": 6,
"attention_heads": 8,
"num_kv_heads": null,
"ffn_dim": 1152,
"activation": "gelu"
},
"text_config": {
"num_layers": 6,
"attention_heads": 8,
"num_kv_heads": null,
"ffn_dim": 1152,
"activation": "silu",
"vocab_size": 32768,
"max_position_embeddings": 194,
"partial_rotary_factor": 0.9,
"rope_theta": 10000.0
},
"hidden_dim": 288,
"layer_norm_eps": 1e-05,
"generate_args": {
"bos_token_id": 1,
"pad_token_id": 2,
"decoder_start_token_id": 1,
"eos_token_id": 2,
"max_length": 194
}
}