hf-transformers-bot commited on
Commit
5f10563
·
verified ·
1 Parent(s): 9cea2eb

Update tiny models for MoshiModel

Browse files
config.json ADDED
@@ -0,0 +1,105 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "architectures": [
3
+ "MoshiModel"
4
+ ],
5
+ "attention_dropout": 0.0,
6
+ "audio_encoder_config": {
7
+ "_frame_rate": null,
8
+ "attention_bias": false,
9
+ "attention_dropout": 0.0,
10
+ "audio_channels": 1,
11
+ "codebook_dim": 64,
12
+ "codebook_size": 64,
13
+ "compress": 2,
14
+ "dilation_growth_rate": 2,
15
+ "head_dim": 16,
16
+ "hidden_act": "gelu",
17
+ "hidden_size": 32,
18
+ "initializer_range": 0.02,
19
+ "intermediate_size": 2048,
20
+ "kernel_size": 7,
21
+ "last_kernel_size": 3,
22
+ "layer_scale_initial_scale": 0.01,
23
+ "max_position_embeddings": 8000,
24
+ "model_type": "mimi",
25
+ "norm_eps": 1e-05,
26
+ "num_attention_heads": 2,
27
+ "num_filters": 8,
28
+ "num_hidden_layers": 2,
29
+ "num_key_value_heads": 2,
30
+ "num_quantizers": 32,
31
+ "num_residual_layers": 1,
32
+ "num_semantic_quantizers": 1,
33
+ "pad_mode": "constant",
34
+ "residual_kernel_size": 3,
35
+ "rope_parameters": {
36
+ "rope_theta": 10000.0,
37
+ "rope_type": "default"
38
+ },
39
+ "sampling_rate": 800,
40
+ "sliding_window": 3,
41
+ "tie_word_embeddings": true,
42
+ "trim_right_ratio": 1.0,
43
+ "upsample_groups": 32,
44
+ "upsampling_ratios": [
45
+ 8,
46
+ 4
47
+ ],
48
+ "use_cache": false,
49
+ "use_causal_conv": true,
50
+ "use_conv_shortcut": false,
51
+ "use_streaming": false,
52
+ "vector_quantization_hidden_dimension": 64
53
+ },
54
+ "audio_vocab_size": 64,
55
+ "bos_token_id": 25,
56
+ "d_ff": 4,
57
+ "depth_decoder_config": {
58
+ "attention_dropout": 0.0,
59
+ "audio_vocab_size": 64,
60
+ "bos_token_id": null,
61
+ "eos_token_id": null,
62
+ "ffn_dim": 16,
63
+ "head_dim": 2,
64
+ "hidden_act": "silu",
65
+ "hidden_size": 16,
66
+ "initializer_range": 0.02,
67
+ "input_size": 32,
68
+ "max_position_embeddings": 5,
69
+ "model_type": "moshi_depth",
70
+ "num_attention_heads": 8,
71
+ "num_codebooks": 4,
72
+ "num_hidden_layers": 2,
73
+ "num_key_value_heads": 8,
74
+ "pad_token_id": null,
75
+ "rms_norm_eps": 1e-08,
76
+ "sliding_window": 4,
77
+ "tie_word_embeddings": false,
78
+ "use_cache": true,
79
+ "vocab_size": 99
80
+ },
81
+ "dtype": "float32",
82
+ "eos_token_id": null,
83
+ "ffn_dim": 32,
84
+ "head_dim": 4,
85
+ "hidden_act": "silu",
86
+ "hidden_size": 32,
87
+ "initializer_range": 0.02,
88
+ "max_position_embeddings": 3000,
89
+ "model_type": "moshi",
90
+ "num_attention_heads": 8,
91
+ "num_codebooks": 4,
92
+ "num_hidden_layers": 2,
93
+ "num_key_value_heads": 8,
94
+ "pad_token_id": 25,
95
+ "rms_norm_eps": 0.001,
96
+ "rope_parameters": {
97
+ "rope_theta": 10000.0,
98
+ "rope_type": "default"
99
+ },
100
+ "sliding_window": 3000,
101
+ "tie_word_embeddings": false,
102
+ "transformers_version": "5.16.0.dev0",
103
+ "use_cache": true,
104
+ "vocab_size": 32000
105
+ }
model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1b7a9a4fe83d445fd506ad0f5016d2aa8b600392b1e69a9a3accb8ae18d39128
3
+ size 4143648
preprocessor_config.json ADDED
@@ -0,0 +1,10 @@
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "chunk_length_s": null,
3
+ "feature_extractor_type": "EncodecFeatureExtractor",
4
+ "feature_size": 1,
5
+ "overlap": null,
6
+ "padding_side": "right",
7
+ "padding_value": 0.0,
8
+ "return_attention_mask": true,
9
+ "sampling_rate": 24000
10
+ }
tokenizer.json ADDED
The diff for this file is too large to render. See raw diff
 
tokenizer_config.json ADDED
@@ -0,0 +1,17 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "backend": "tokenizers",
3
+ "bos_token_id": null,
4
+ "chat_template": null,
5
+ "clean_up_tokenization_spaces": false,
6
+ "eos_token_id": null,
7
+ "is_local": true,
8
+ "local_files_only": false,
9
+ "model_input_names": [
10
+ "input_ids",
11
+ "attention_mask"
12
+ ],
13
+ "model_max_length": 3000,
14
+ "pad_token_id": null,
15
+ "tokenizer_class": "TokenizersBackend",
16
+ "unk_token": "<unk>"
17
+ }