File size: 4,071 Bytes
a7c2def |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 |
{
"module": "keras_hub.src.models.moonshine.moonshine_audio_to_text",
"class_name": "MoonshineAudioToText",
"config": {
"backbone": {
"module": "keras_hub.src.models.moonshine.moonshine_backbone",
"class_name": "MoonshineBackbone",
"config": {
"name": "moonshine_backbone_1",
"trainable": true,
"vocabulary_size": 32768,
"filter_dim": 288,
"encoder_num_layers": 6,
"decoder_num_layers": 6,
"hidden_dim": 288,
"intermediate_dim": 1152,
"encoder_num_heads": 8,
"decoder_num_heads": 8,
"feedforward_expansion_factor": 4,
"encoder_use_swiglu_activation": false,
"decoder_use_swiglu_activation": true,
"max_position_embeddings": 194,
"pad_head_dim_to_multiple_of": null,
"partial_rotary_factor": 0.9,
"dropout": 0.0,
"initializer_range": 0.02,
"rope_theta": 10000.0,
"attention_bias": false,
"attention_dropout": 0.0,
"dtype": "float32"
},
"registered_name": "keras_hub>MoonshineBackbone"
},
"preprocessor": {
"module": "keras_hub.src.models.moonshine.moonshine_audio_to_text_preprocessor",
"class_name": "MoonshineAudioToTextPreprocessor",
"config": {
"name": "moonshine_audio_to_text_preprocessor_1",
"trainable": true,
"dtype": {
"module": "keras",
"class_name": "DTypePolicy",
"config": {
"name": "float32"
},
"registered_name": null
},
"tokenizer": {
"module": "keras_hub.src.models.moonshine.moonshine_tokenizer",
"class_name": "MoonshineTokenizer",
"config": {
"name": "moonshine_tokenizer_1",
"trainable": true,
"dtype": {
"module": "keras",
"class_name": "DTypePolicy",
"config": {
"name": "int32"
},
"registered_name": null
},
"config_file": "tokenizer.json",
"proto": null,
"sequence_length": null,
"add_bos": false,
"add_eos": false
},
"registered_name": "keras_hub>MoonshineTokenizer"
},
"audio_converter": {
"module": "keras_hub.src.models.moonshine.moonshine_audio_converter",
"class_name": "MoonshineAudioConverter",
"config": {
"name": "moonshine_audio_converter_1",
"trainable": true,
"dtype": {
"module": "keras",
"class_name": "DTypePolicy",
"config": {
"name": "float32"
},
"registered_name": null
},
"sampling_rate": 16000,
"padding_value": 0.0,
"do_normalize": false
},
"registered_name": "keras_hub>MoonshineAudioConverter"
},
"config_file": "preprocessor.json"
},
"registered_name": "keras_hub>MoonshineAudioToTextPreprocessor"
},
"name": "moonshine_audio_to_text_1"
},
"registered_name": "keras_hub>MoonshineAudioToText"
} |