moonshine_base / kf_config.json
IMvision12's picture
Upgrade kf_config.json to v2 (nested config + weight_dtype)
f19deea verified
Raw
History Blame Contribute Delete
947 Bytes
{
"library_name": "kerasformers",
"kerasformers_version": "1.2.1",
"model_module": "kerasformers.models.moonshine",
"model_class": "MoonshineModel",
"variant": "moonshine_base",
"weights": "model.weights.h5",
"schema_version": 2,
"weight_dtype": "float32",
"model_type": "moonshine",
"audio_config": {
"num_layers": 8,
"attention_heads": 8,
"num_kv_heads": null,
"ffn_dim": 1664,
"activation": "gelu"
},
"text_config": {
"num_layers": 8,
"attention_heads": 8,
"num_kv_heads": null,
"ffn_dim": 1664,
"activation": "silu",
"vocab_size": 32768,
"max_position_embeddings": 194,
"partial_rotary_factor": 0.62,
"rope_theta": 10000.0
},
"hidden_dim": 416,
"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
}
}