tiny-random-MoshiModel / config.json
hf-transformers-bot's picture
Update tiny models for MoshiModel
5f10563 verified
Raw
History Blame Contribute Delete
2.59 kB
{
"architectures": [
"MoshiModel"
],
"attention_dropout": 0.0,
"audio_encoder_config": {
"_frame_rate": null,
"attention_bias": false,
"attention_dropout": 0.0,
"audio_channels": 1,
"codebook_dim": 64,
"codebook_size": 64,
"compress": 2,
"dilation_growth_rate": 2,
"head_dim": 16,
"hidden_act": "gelu",
"hidden_size": 32,
"initializer_range": 0.02,
"intermediate_size": 2048,
"kernel_size": 7,
"last_kernel_size": 3,
"layer_scale_initial_scale": 0.01,
"max_position_embeddings": 8000,
"model_type": "mimi",
"norm_eps": 1e-05,
"num_attention_heads": 2,
"num_filters": 8,
"num_hidden_layers": 2,
"num_key_value_heads": 2,
"num_quantizers": 32,
"num_residual_layers": 1,
"num_semantic_quantizers": 1,
"pad_mode": "constant",
"residual_kernel_size": 3,
"rope_parameters": {
"rope_theta": 10000.0,
"rope_type": "default"
},
"sampling_rate": 800,
"sliding_window": 3,
"tie_word_embeddings": true,
"trim_right_ratio": 1.0,
"upsample_groups": 32,
"upsampling_ratios": [
8,
4
],
"use_cache": false,
"use_causal_conv": true,
"use_conv_shortcut": false,
"use_streaming": false,
"vector_quantization_hidden_dimension": 64
},
"audio_vocab_size": 64,
"bos_token_id": 25,
"d_ff": 4,
"depth_decoder_config": {
"attention_dropout": 0.0,
"audio_vocab_size": 64,
"bos_token_id": null,
"eos_token_id": null,
"ffn_dim": 16,
"head_dim": 2,
"hidden_act": "silu",
"hidden_size": 16,
"initializer_range": 0.02,
"input_size": 32,
"max_position_embeddings": 5,
"model_type": "moshi_depth",
"num_attention_heads": 8,
"num_codebooks": 4,
"num_hidden_layers": 2,
"num_key_value_heads": 8,
"pad_token_id": null,
"rms_norm_eps": 1e-08,
"sliding_window": 4,
"tie_word_embeddings": false,
"use_cache": true,
"vocab_size": 99
},
"dtype": "float32",
"eos_token_id": null,
"ffn_dim": 32,
"head_dim": 4,
"hidden_act": "silu",
"hidden_size": 32,
"initializer_range": 0.02,
"max_position_embeddings": 3000,
"model_type": "moshi",
"num_attention_heads": 8,
"num_codebooks": 4,
"num_hidden_layers": 2,
"num_key_value_heads": 8,
"pad_token_id": 25,
"rms_norm_eps": 0.001,
"rope_parameters": {
"rope_theta": 10000.0,
"rope_type": "default"
},
"sliding_window": 3000,
"tie_word_embeddings": false,
"transformers_version": "5.16.0.dev0",
"use_cache": true,
"vocab_size": 32000
}