REALBITS commited on
Commit
8645e25
·
verified ·
1 Parent(s): 442f906

Initial M16 upload — fp16-quantized MOSS-TTS-Nano LM (shared external data)

Browse files
.gitattributes CHANGED
@@ -33,3 +33,5 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
33
  *.zip filter=lfs diff=lfs merge=lfs -text
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
 
 
 
33
  *.zip filter=lfs diff=lfs merge=lfs -text
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
36
+ moss_tts_global_shared_fp16.data filter=lfs diff=lfs merge=lfs -text
37
+ moss_tts_local_fixed_sampled_frame_fp16.data filter=lfs diff=lfs merge=lfs -text
browser_poc_manifest.json ADDED
The diff for this file is too large to render. See raw diff
 
moss_tts_decode_step.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a737ab131f559b026565fe5eacaa83483ba96bb548989337e4bafab4a619a938
3
+ size 502789
moss_tts_global_shared_fp16.data ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:604741414938c22d27ad8f930313cac50e322bcc3dec846183d38baf39272931
3
+ size 220445184
moss_tts_local_fixed_sampled_frame.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a4022ecc69ae4b0bc1ff6e28bf3e0da3b451146144a9e74ae1c8cb8cb394c1a0
3
+ size 819138
moss_tts_local_fixed_sampled_frame_fp16.data ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ce4ee4572ac32d57cd485292c97607a5f093d6388e49fe1a6582e0d10eab6814
3
+ size 113270784
moss_tts_prefill.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:33f64c445fe3be504ae7f407504caa76b39a83d4e972e0c1e6165c5da526d784
3
+ size 475866
tokenizer.model ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c353ee1479b536bf414c1b247f5542b6607fb8ae91320e5af1781fee200fddff
3
+ size 470897
tts_browser_onnx_meta.json ADDED
@@ -0,0 +1,186 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "format_version": 1,
3
+ "checkpoint_path": "MOSS-TTS-Nano",
4
+ "files": {
5
+ "prefill": "moss_tts_prefill.onnx",
6
+ "decode_step": "moss_tts_decode_step.onnx",
7
+ "local_decoder": "moss_tts_local_decoder.onnx",
8
+ "local_cached_step": "moss_tts_local_cached_step.onnx",
9
+ "local_fixed_sampled_frame": "moss_tts_local_fixed_sampled_frame.onnx"
10
+ },
11
+ "external_data_files": {
12
+ "moss_tts_prefill.onnx": [
13
+ "moss_tts_global_shared.data"
14
+ ],
15
+ "moss_tts_decode_step.onnx": [
16
+ "moss_tts_global_shared.data"
17
+ ],
18
+ "moss_tts_local_decoder.onnx": [
19
+ "moss_tts_local_shared.data"
20
+ ],
21
+ "moss_tts_local_cached_step.onnx": [
22
+ "moss_tts_local_shared.data"
23
+ ],
24
+ "moss_tts_local_fixed_sampled_frame.onnx": [
25
+ "moss_tts_local_shared.data"
26
+ ]
27
+ },
28
+ "model_config": {
29
+ "n_vq": 16,
30
+ "row_width": 17,
31
+ "hidden_size": 768,
32
+ "global_layers": 12,
33
+ "global_heads": 12,
34
+ "head_dim": 64,
35
+ "local_layers": 1,
36
+ "local_heads": 12,
37
+ "local_head_dim": 64,
38
+ "vocab_size": 16384,
39
+ "audio_codebook_sizes": [
40
+ 1024,
41
+ 1024,
42
+ 1024,
43
+ 1024,
44
+ 1024,
45
+ 1024,
46
+ 1024,
47
+ 1024,
48
+ 1024,
49
+ 1024,
50
+ 1024,
51
+ 1024,
52
+ 1024,
53
+ 1024,
54
+ 1024,
55
+ 1024
56
+ ],
57
+ "audio_pad_token_id": 1024,
58
+ "pad_token_id": 3,
59
+ "im_start_token_id": 4,
60
+ "im_end_token_id": 5,
61
+ "audio_start_token_id": 6,
62
+ "audio_end_token_id": 7,
63
+ "audio_user_slot_token_id": 8,
64
+ "audio_assistant_slot_token_id": 9
65
+ },
66
+ "onnx": {
67
+ "opset": 17,
68
+ "prefill_output_names": [
69
+ "global_hidden",
70
+ "present_key_0",
71
+ "present_value_0",
72
+ "present_key_1",
73
+ "present_value_1",
74
+ "present_key_2",
75
+ "present_value_2",
76
+ "present_key_3",
77
+ "present_value_3",
78
+ "present_key_4",
79
+ "present_value_4",
80
+ "present_key_5",
81
+ "present_value_5",
82
+ "present_key_6",
83
+ "present_value_6",
84
+ "present_key_7",
85
+ "present_value_7",
86
+ "present_key_8",
87
+ "present_value_8",
88
+ "present_key_9",
89
+ "present_value_9",
90
+ "present_key_10",
91
+ "present_value_10",
92
+ "present_key_11",
93
+ "present_value_11"
94
+ ],
95
+ "decode_input_names": [
96
+ "input_ids",
97
+ "past_valid_lengths",
98
+ "past_key_0",
99
+ "past_value_0",
100
+ "past_key_1",
101
+ "past_value_1",
102
+ "past_key_2",
103
+ "past_value_2",
104
+ "past_key_3",
105
+ "past_value_3",
106
+ "past_key_4",
107
+ "past_value_4",
108
+ "past_key_5",
109
+ "past_value_5",
110
+ "past_key_6",
111
+ "past_value_6",
112
+ "past_key_7",
113
+ "past_value_7",
114
+ "past_key_8",
115
+ "past_value_8",
116
+ "past_key_9",
117
+ "past_value_9",
118
+ "past_key_10",
119
+ "past_value_10",
120
+ "past_key_11",
121
+ "past_value_11"
122
+ ],
123
+ "decode_output_names": [
124
+ "global_hidden",
125
+ "present_key_0",
126
+ "present_value_0",
127
+ "present_key_1",
128
+ "present_value_1",
129
+ "present_key_2",
130
+ "present_value_2",
131
+ "present_key_3",
132
+ "present_value_3",
133
+ "present_key_4",
134
+ "present_value_4",
135
+ "present_key_5",
136
+ "present_value_5",
137
+ "present_key_6",
138
+ "present_value_6",
139
+ "present_key_7",
140
+ "present_value_7",
141
+ "present_key_8",
142
+ "present_value_8",
143
+ "present_key_9",
144
+ "present_value_9",
145
+ "present_key_10",
146
+ "present_value_10",
147
+ "present_key_11",
148
+ "present_value_11"
149
+ ],
150
+ "local_cached_input_names": [
151
+ "global_hidden",
152
+ "text_token_id",
153
+ "audio_token_id",
154
+ "channel_index",
155
+ "step_type",
156
+ "past_valid_lengths",
157
+ "local_past_key_0",
158
+ "local_past_value_0"
159
+ ],
160
+ "local_cached_output_names": [
161
+ "text_logits",
162
+ "audio_logits",
163
+ "local_present_key_0",
164
+ "local_present_value_0"
165
+ ],
166
+ "local_fixed_sampled_frame_input_names": [
167
+ "global_hidden",
168
+ "repetition_seen_mask",
169
+ "assistant_random_u",
170
+ "audio_random_u"
171
+ ],
172
+ "local_fixed_sampled_frame_output_names": [
173
+ "should_continue",
174
+ "frame_token_ids"
175
+ ],
176
+ "fixed_sampled_frame_constants": {
177
+ "text_temperature": 1.0,
178
+ "text_top_p": 1.0,
179
+ "text_top_k": 50,
180
+ "audio_temperature": 0.8,
181
+ "audio_top_p": 0.95,
182
+ "audio_top_k": 25,
183
+ "audio_repetition_penalty": 1.2
184
+ }
185
+ }
186
+ }