AEmotionStudio commited on
Commit
30b408a
·
verified ·
1 Parent(s): 90b2880

mirror: soar ← rednote-hilab/dots.tts-soar

Browse files
.gitattributes CHANGED
@@ -34,3 +34,4 @@ saved_model/**/* 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
  base/tokenizer.json 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
  base/tokenizer.json filter=lfs diff=lfs merge=lfs -text
37
+ soar/tokenizer.json filter=lfs diff=lfs merge=lfs -text
soar/.gitattributes ADDED
@@ -0,0 +1,36 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ *.7z filter=lfs diff=lfs merge=lfs -text
2
+ *.arrow filter=lfs diff=lfs merge=lfs -text
3
+ *.bin filter=lfs diff=lfs merge=lfs -text
4
+ *.bz2 filter=lfs diff=lfs merge=lfs -text
5
+ *.ckpt filter=lfs diff=lfs merge=lfs -text
6
+ *.ftz filter=lfs diff=lfs merge=lfs -text
7
+ *.gz filter=lfs diff=lfs merge=lfs -text
8
+ *.h5 filter=lfs diff=lfs merge=lfs -text
9
+ *.joblib filter=lfs diff=lfs merge=lfs -text
10
+ *.lfs.* filter=lfs diff=lfs merge=lfs -text
11
+ *.mlmodel filter=lfs diff=lfs merge=lfs -text
12
+ *.model filter=lfs diff=lfs merge=lfs -text
13
+ *.msgpack filter=lfs diff=lfs merge=lfs -text
14
+ *.npy filter=lfs diff=lfs merge=lfs -text
15
+ *.npz filter=lfs diff=lfs merge=lfs -text
16
+ *.onnx filter=lfs diff=lfs merge=lfs -text
17
+ *.ot filter=lfs diff=lfs merge=lfs -text
18
+ *.parquet filter=lfs diff=lfs merge=lfs -text
19
+ *.pb filter=lfs diff=lfs merge=lfs -text
20
+ *.pickle filter=lfs diff=lfs merge=lfs -text
21
+ *.pkl filter=lfs diff=lfs merge=lfs -text
22
+ *.pt filter=lfs diff=lfs merge=lfs -text
23
+ *.pth filter=lfs diff=lfs merge=lfs -text
24
+ *.rar filter=lfs diff=lfs merge=lfs -text
25
+ *.safetensors filter=lfs diff=lfs merge=lfs -text
26
+ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
27
+ *.tar.* filter=lfs diff=lfs merge=lfs -text
28
+ *.tar filter=lfs diff=lfs merge=lfs -text
29
+ *.tflite filter=lfs diff=lfs merge=lfs -text
30
+ *.tgz filter=lfs diff=lfs merge=lfs -text
31
+ *.wasm filter=lfs diff=lfs merge=lfs -text
32
+ *.xz 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
+ tokenizer.json filter=lfs diff=lfs merge=lfs -text
soar/README.md ADDED
@@ -0,0 +1,181 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: apache-2.0
3
+ pipeline_tag: text-to-speech
4
+ tags:
5
+ - text-to-speech
6
+ - tts
7
+ - audio
8
+ - speech-synthesis
9
+ - voice-cloning
10
+ - autoregressive
11
+ - flow-matching
12
+ - post-trained
13
+ library_name: dots_tts
14
+ base_model: rednote-hilab/dots.tts-base
15
+ ---
16
+
17
+ # dots.tts-soar
18
+
19
+ <p align="left">
20
+ <a href="https://github.com/rednote-hilab/dots.tts"><img src="https://img.shields.io/badge/GitHub-rednote--hilab%2Fdots.tts-blue?logo=github" alt="GitHub"></a>
21
+ <a href="https://huggingface.co/spaces/rednote-hilab/dots.tts"><img src="https://img.shields.io/badge/%F0%9F%A4%97%20Spaces-Playground-orange" alt="Playground"></a>
22
+ <a href="https://rednote-hilab.github.io/dots.tts-demo/"><img src="https://img.shields.io/badge/Demo%20Page-Live-red" alt="Demo Page"></a>
23
+ </p>
24
+
25
+ **dots.tts** is a **2B-parameter fully continuous, end-to-end autoregressive (AR) text-to-speech system**. The backbone pairs a semantic encoder, an LLM, and an autoregressive flow-matching acoustic head over a 48 kHz AudioVAE — no discrete codec tokens anywhere in the pipeline.
26
+
27
+ This repository hosts **`dots.tts-soar`** — the pretrained backbone further refined with **Self-corrective Alignment (SCA)**, a reward-free flow-matching-native post-training stage. SCA pushes the model to the **highest zero-shot fidelity and speaker similarity** of the three releases and is the **recommended default for production zero-shot voice cloning**.
28
+
29
+ <table>
30
+ <tr>
31
+ <td align="left" valign="middle"><a href="https://huggingface.co/rednote-hilab/dots.tts-base"><img src="https://img.shields.io/badge/%F0%9F%A4%97%20Hugging%20Face-dots.tts--base-yellow" alt="dots.tts-base"></a></td>
32
+ <td>Pretrain (~1.5M h). Fine-tuning, full CFG / NFE control.</td>
33
+ </tr>
34
+ <tr>
35
+ <td align="left" valign="middle"><a href="https://huggingface.co/rednote-hilab/dots.tts-soar"><img src="https://img.shields.io/badge/%F0%9F%A4%97%20Hugging%20Face-dots.tts--soar-yellow" alt="dots.tts-soar"></a></td>
36
+ <td>← <em>you are here</em> — + Self-corrective Alignment. <strong>Highest zero-shot fidelity and speaker similarity</strong>; also recommended for fine-tuning.</td>
37
+ </tr>
38
+ <tr>
39
+ <td align="left" valign="middle"><a href="https://huggingface.co/rednote-hilab/dots.tts-mf"><img src="https://img.shields.io/badge/%F0%9F%A4%97%20Hugging%20Face-dots.tts--mf-yellow" alt="dots.tts-mf"></a></td>
40
+ <td>+ MeanFlow distillation. Few-step inference (NFE = 4), low latency.</td>
41
+ </tr>
42
+ </table>
43
+
44
+ ---
45
+
46
+ ## Quick Start
47
+
48
+ ### Installation
49
+
50
+ ```bash
51
+ conda create -n dots_tts python=3.10 -y
52
+ conda activate dots_tts
53
+
54
+ python -m pip install --upgrade pip
55
+ python -m pip install "git+https://github.com/rednote-hilab/dots.tts.git" \
56
+ -c "https://raw.githubusercontent.com/rednote-hilab/dots.tts/main/constraints/recommended.txt"
57
+ ```
58
+
59
+ ### CLI
60
+
61
+ ```bash
62
+ # Continuation voice cloning (reference audio + transcript) — recommended
63
+ dots.tts \
64
+ --model-name-or-path rednote-hilab/dots.tts-soar \
65
+ --text "Hello, this is a zero-shot voice cloning demonstration." \
66
+ --prompt-audio /path/to/reference.wav \
67
+ --prompt-text "The exact transcript of the reference audio." \
68
+ --output clone.wav
69
+ ```
70
+
71
+ ### Python API
72
+
73
+ ```python
74
+ from dots_tts.runtime import DotsTtsRuntime
75
+ import soundfile as sf
76
+
77
+ runtime = DotsTtsRuntime.from_pretrained(
78
+ "rednote-hilab/dots.tts-soar",
79
+ precision="bfloat16",
80
+ )
81
+
82
+ result = runtime.generate(
83
+ text="Hello, this is a quick speech synthesis test.",
84
+ prompt_audio_path="/path/to/reference.wav",
85
+ prompt_text="The exact transcript of the reference audio.",
86
+ num_steps=10,
87
+ guidance_scale=1.2,
88
+ )
89
+
90
+ sf.write("output.wav", result["audio"].float().cpu().squeeze().numpy(), result["sample_rate"])
91
+ ```
92
+
93
+ ### Recommended sampling settings
94
+
95
+ | Flag | Recommended | Notes |
96
+ |---|---:|---|
97
+ | `--num-steps` | `10`–`32` | Flow-matching sampling steps; higher = better quality, slower |
98
+ | `--guidance-scale` | `1.2` (default) | Standard CFG; SCA already tightens text and timbre adherence so small CFG suffices |
99
+
100
+ ### Fine-tuning
101
+
102
+ Both `dots.tts-base` and `dots.tts-soar` are valid fine-tuning starting points. Pick `dots.tts-soar` when you want to inherit its tightened text/timbre alignment on top of the pretrained backbone. See the [training script](https://github.com/rednote-hilab/dots.tts/blob/main/scripts/train_dots_tts.py) and [smoke config](https://github.com/rednote-hilab/dots.tts/blob/main/configs/dots_tts.yaml) in the source repository:
103
+
104
+ ```bash
105
+ accelerate launch scripts/train_dots_tts.py --config configs/dots_tts.yaml
106
+ ```
107
+
108
+ ---
109
+
110
+ ## Architecture
111
+
112
+ A frozen **AudioVAE** encodes 48 kHz mono waveform into a continuous latent and decodes it back via a BigVGAN-style causal decoder. An **autoregressive backbone** predicts that latent one patch at a time:
113
+
114
+ - **Semantic encoder** — re-encodes each newly generated VAE patch into a compact embedding for the LLM, stripping high-variance acoustic detail.
115
+ - **LLM** — initialized from **Qwen2.5-1.5B-Base**, consumes BPE text directly (no phonemes), emits one hidden state per audio step.
116
+ - **AR flow-matching head** — a DiT that conditions on the LLM hidden state and the AR prefix to denoise the next VAE patch, with a frozen CAM++ speaker x-vector as side input.
117
+
118
+ **Self-corrective Alignment** is a reward-free, flow-matching-native post-training stage applied on top of `dots.tts-base`. It improves text and speaker adherence without changing inference cost or sampling schedule.
119
+
120
+ ---
121
+
122
+ ## Performance — `dots.tts-soar`
123
+
124
+ ### Seed-TTS-Eval — **state-of-the-art average SIM (79.2)**
125
+
126
+ | Model | Params | test-en WER↓ / SIM↑ | test-zh WER↓ / SIM↑ | test-zh-hard WER↓ / SIM↑ | **Avg WER↓ / SIM↑** |
127
+ |---|---:|:---:|:---:|:---:|:---:|
128
+ | Seed-TTS | — | 2.25 / 76.2 | 1.12 / 79.6 | 7.59 / 77.6 | 3.65 / 77.8 |
129
+ | Qwen3-TTS | 1.7B | **1.23** / 71.7 | 1.22 / 77.0 | 6.76 / 74.8 | 3.07 / 74.5 |
130
+ | VoxCPM 2 | 2B | 1.84 / 75.3 | 0.97 / 79.5 | 8.13 / 75.3 | 3.65 / 76.7 |
131
+ | dots.tts-base | 2B | 1.34 / 76.8 | 0.96 / 80.5 | **6.46** / 79.2 | **2.92** / 78.8 |
132
+ | **dots.tts-soar** | **2B** | 1.30 / **77.1** | **0.94** / **81.0** | 6.60 / **79.5** | 2.95 / **79.2** |
133
+
134
+ ### MiniMax Multilingual — **highest average SIM (83.9) across 24 languages**
135
+
136
+ | Model | Avg WER↓ | Avg SIM↑ |
137
+ |---|:---:|:---:|
138
+ | MiniMax | **2.8** | 76.6 |
139
+ | Fish-Audio S2 | 3.7 | 78.0 |
140
+ | VoxCPM 2 | 5.7 | 82.3 |
141
+ | dots.tts-base | 6.6 | 83.5 |
142
+ | **dots.tts-soar** | 6.8 | **83.9** |
143
+
144
+ ### CV3-Eval — **leads both cross-lingual SIM subsets**
145
+
146
+ | Model | en→zh SIM↑ | zh→en SIM↑ |
147
+ |---|:---:|:---:|
148
+ | CosyVoice 3 (1.5B) | 66.9 | 66.4 |
149
+ | dots.tts-base | 74.6 | 71.9 |
150
+ | **dots.tts-soar** | **75.0** | **72.8** |
151
+
152
+ ### EmergentTTS-Eval — **top Syntactic Complexity in the table (65.7%)**
153
+
154
+ On head-to-head judging vs. `gpt-4o-mini-tts`, `dots.tts-soar` posts **65.7% on Syntactic Complexity — above every closed-source system listed**, while keeping competitive Emotions / Questions scores.
155
+
156
+ See the [project README](https://github.com/rednote-hilab/dots.tts#-performance) for full benchmark tables.
157
+
158
+ ---
159
+
160
+ ## Risks and Limitations
161
+
162
+ - **Misuse risk.** High-fidelity zero-shot voice cloning can produce highly realistic synthetic speech. This checkpoint is intended for research and authorized deployment. Do **not** use it for impersonation, fraud, or disinformation. Combine downstream use with consent-aware reference-audio policies, robust synthetic-speech detection, and content watermarking. Clearly mark AI-generated audio.
163
+ - **Low-resource WER gap.** A BPE backbone inherits the text LLM's language coverage at the cost of a higher data appetite. On script-divergent and under-represented languages (Arabic, Hindi, Turkish, Vietnamese) WER is higher than on high-resource languages; speaker similarity is preserved.
164
+ - **Speech-heavy training.** The backbone is trained on a speech-heavy mixture. Singing and unified speech + sound generation are not covered.
165
+
166
+ ---
167
+
168
+ ## Citation
169
+
170
+ ```bibtex
171
+ @article{dotstts2026,
172
+ title = {dots.tts Technical Report},
173
+ author = {dots.tts Team},
174
+ journal = {arXiv preprint},
175
+ year = {2026},
176
+ }
177
+ ```
178
+
179
+ ## License
180
+
181
+ Released under [Apache-2.0](https://www.apache.org/licenses/LICENSE-2.0).
soar/added_tokens.json ADDED
@@ -0,0 +1,31 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "</tool_call>": 151658,
3
+ "<tool_call>": 151657,
4
+ "<|audio_comp_end|>": 151667,
5
+ "<|audio_comp_span|>": 151666,
6
+ "<|audio_comp_start|>": 151665,
7
+ "<|audio_gen_end|>": 151670,
8
+ "<|audio_gen_span|>": 151669,
9
+ "<|audio_gen_start|>": 151668,
10
+ "<|box_end|>": 151649,
11
+ "<|box_start|>": 151648,
12
+ "<|endoftext|>": 151643,
13
+ "<|file_sep|>": 151664,
14
+ "<|fim_middle|>": 151660,
15
+ "<|fim_pad|>": 151662,
16
+ "<|fim_prefix|>": 151659,
17
+ "<|fim_suffix|>": 151661,
18
+ "<|im_end|>": 151645,
19
+ "<|im_start|>": 151644,
20
+ "<|image_pad|>": 151655,
21
+ "<|object_ref_end|>": 151647,
22
+ "<|object_ref_start|>": 151646,
23
+ "<|quad_end|>": 151651,
24
+ "<|quad_start|>": 151650,
25
+ "<|repo_name|>": 151663,
26
+ "<|text_cond_end|>": 151671,
27
+ "<|video_pad|>": 151656,
28
+ "<|vision_end|>": 151653,
29
+ "<|vision_pad|>": 151654,
30
+ "<|vision_start|>": 151652
31
+ }
soar/chat_template.jinja ADDED
@@ -0,0 +1,54 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {%- if tools %}
2
+ {{- '<|im_start|>system\n' }}
3
+ {%- if messages[0]['role'] == 'system' %}
4
+ {{- messages[0]['content'] }}
5
+ {%- else %}
6
+ {{- 'You are a helpful assistant.' }}
7
+ {%- endif %}
8
+ {{- "\n\n# Tools\n\nYou may call one or more functions to assist with the user query.\n\nYou are provided with function signatures within <tools></tools> XML tags:\n<tools>" }}
9
+ {%- for tool in tools %}
10
+ {{- "\n" }}
11
+ {{- tool | tojson }}
12
+ {%- endfor %}
13
+ {{- "\n</tools>\n\nFor each function call, return a json object with function name and arguments within <tool_call></tool_call> XML tags:\n<tool_call>\n{\"name\": <function-name>, \"arguments\": <args-json-object>}\n</tool_call><|im_end|>\n" }}
14
+ {%- else %}
15
+ {%- if messages[0]['role'] == 'system' %}
16
+ {{- '<|im_start|>system\n' + messages[0]['content'] + '<|im_end|>\n' }}
17
+ {%- else %}
18
+ {{- '<|im_start|>system\nYou are a helpful assistant.<|im_end|>\n' }}
19
+ {%- endif %}
20
+ {%- endif %}
21
+ {%- for message in messages %}
22
+ {%- if (message.role == "user") or (message.role == "system" and not loop.first) or (message.role == "assistant" and not message.tool_calls) %}
23
+ {{- '<|im_start|>' + message.role + '\n' + message.content + '<|im_end|>' + '\n' }}
24
+ {%- elif message.role == "assistant" %}
25
+ {{- '<|im_start|>' + message.role }}
26
+ {%- if message.content %}
27
+ {{- '\n' + message.content }}
28
+ {%- endif %}
29
+ {%- for tool_call in message.tool_calls %}
30
+ {%- if tool_call.function is defined %}
31
+ {%- set tool_call = tool_call.function %}
32
+ {%- endif %}
33
+ {{- '\n<tool_call>\n{"name": "' }}
34
+ {{- tool_call.name }}
35
+ {{- '", "arguments": ' }}
36
+ {{- tool_call.arguments | tojson }}
37
+ {{- '}\n</tool_call>' }}
38
+ {%- endfor %}
39
+ {{- '<|im_end|>\n' }}
40
+ {%- elif message.role == "tool" %}
41
+ {%- if (loop.index0 == 0) or (messages[loop.index0 - 1].role != "tool") %}
42
+ {{- '<|im_start|>user' }}
43
+ {%- endif %}
44
+ {{- '\n<tool_response>\n' }}
45
+ {{- message.content }}
46
+ {{- '\n</tool_response>' }}
47
+ {%- if loop.last or (messages[loop.index0 + 1].role != "tool") %}
48
+ {{- '<|im_end|>\n' }}
49
+ {%- endif %}
50
+ {%- endif %}
51
+ {%- endfor %}
52
+ {%- if add_generation_prompt %}
53
+ {{- '<|im_start|>assistant\n' }}
54
+ {%- endif %}
soar/config.json ADDED
@@ -0,0 +1,113 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "model_type": "dots_tts",
3
+ "latent_dim": 128,
4
+ "patch_size": 4,
5
+ "cfg_droprate": 0.2,
6
+ "PatchEncoder": {
7
+ "num_layers": 24,
8
+ "num_heads": 16,
9
+ "hidden_size": 1024,
10
+ "ffn_hidden_size": 4096,
11
+ "modulation": false,
12
+ "qkv_bias": false,
13
+ "qk_norm": true,
14
+ "attn_dropout": 0.0,
15
+ "dropout": 0.1,
16
+ "norm_layer": "RMSNorm",
17
+ "alibi_bias": false,
18
+ "rotary_bias": true,
19
+ "rotary_theta": 10000.0,
20
+ "input_dim": 128,
21
+ "causal": true
22
+ },
23
+ "DiT": {
24
+ "num_layers": 18,
25
+ "num_heads": 16,
26
+ "hidden_size": 1024,
27
+ "ffn_hidden_size": 4096,
28
+ "modulation": true,
29
+ "qkv_bias": false,
30
+ "qk_norm": true,
31
+ "attn_dropout": 0.0,
32
+ "dropout": 0.0,
33
+ "norm_layer": "RMSNorm",
34
+ "alibi_bias": false,
35
+ "rotary_bias": true,
36
+ "rotary_theta": 10000.0
37
+ },
38
+ "vocoder": {
39
+ "sample_rate": 48000,
40
+ "upsample_rates": [
41
+ 10,
42
+ 6,
43
+ 4,
44
+ 2,
45
+ 2,
46
+ 2
47
+ ],
48
+ "upsample_kernel_sizes": [
49
+ 20,
50
+ 12,
51
+ 8,
52
+ 4,
53
+ 4,
54
+ 4
55
+ ],
56
+ "upsample_initial_channel": 1536,
57
+ "resblock": "1",
58
+ "resblock_kernel_sizes": [
59
+ 3,
60
+ 7,
61
+ 11
62
+ ],
63
+ "resblock_dilation_sizes": [
64
+ [
65
+ 1,
66
+ 3,
67
+ 5
68
+ ],
69
+ [
70
+ 1,
71
+ 3,
72
+ 5
73
+ ],
74
+ [
75
+ 1,
76
+ 3,
77
+ 5
78
+ ]
79
+ ],
80
+ "downsample_rates": [
81
+ 2,
82
+ 2,
83
+ 2,
84
+ 4,
85
+ 6,
86
+ 10
87
+ ],
88
+ "downsample_channels": [
89
+ 12,
90
+ 24,
91
+ 48,
92
+ 96,
93
+ 192,
94
+ 384,
95
+ 768
96
+ ],
97
+ "activation": "snakebeta",
98
+ "snake_logscale": true,
99
+ "latent_dim": 128,
100
+ "causal": true,
101
+ "mi_num_layers": 4,
102
+ "causal_encoder": true,
103
+ "use_bias_at_final": false,
104
+ "use_tanh_at_final": false
105
+ },
106
+ "fm_sigma": 0.0,
107
+ "xvec_drop_rate": 0.2,
108
+ "campplus_embedding_size": 512,
109
+ "xvec_max_audio_seconds": 10.0,
110
+ "architectures": [
111
+ "DotsTTSForConditionalGeneration"
112
+ ]
113
+ }
soar/latent_stats.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:313b13af56d659ecf869d5f854508fcf823c8f957aefc6bc05244991abd6ffe1
3
+ size 3197
soar/llm_config.json ADDED
@@ -0,0 +1,114 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "vocab_size": 151672,
3
+ "max_position_embeddings": 131072,
4
+ "hidden_size": 1536,
5
+ "intermediate_size": 8960,
6
+ "num_hidden_layers": 28,
7
+ "num_attention_heads": 12,
8
+ "use_sliding_window": false,
9
+ "sliding_window": null,
10
+ "max_window_layers": 28,
11
+ "num_key_value_heads": 2,
12
+ "hidden_act": "silu",
13
+ "initializer_range": 0.02,
14
+ "rms_norm_eps": 1e-06,
15
+ "use_cache": true,
16
+ "rope_theta": 1000000.0,
17
+ "rope_scaling": null,
18
+ "attention_dropout": 0.0,
19
+ "layer_types": [
20
+ "full_attention",
21
+ "full_attention",
22
+ "full_attention",
23
+ "full_attention",
24
+ "full_attention",
25
+ "full_attention",
26
+ "full_attention",
27
+ "full_attention",
28
+ "full_attention",
29
+ "full_attention",
30
+ "full_attention",
31
+ "full_attention",
32
+ "full_attention",
33
+ "full_attention",
34
+ "full_attention",
35
+ "full_attention",
36
+ "full_attention",
37
+ "full_attention",
38
+ "full_attention",
39
+ "full_attention",
40
+ "full_attention",
41
+ "full_attention",
42
+ "full_attention",
43
+ "full_attention",
44
+ "full_attention",
45
+ "full_attention",
46
+ "full_attention",
47
+ "full_attention"
48
+ ],
49
+ "return_dict": true,
50
+ "output_hidden_states": false,
51
+ "torchscript": false,
52
+ "dtype": "bfloat16",
53
+ "pruned_heads": {},
54
+ "tie_word_embeddings": true,
55
+ "chunk_size_feed_forward": 0,
56
+ "is_encoder_decoder": false,
57
+ "is_decoder": false,
58
+ "cross_attention_hidden_size": null,
59
+ "add_cross_attention": false,
60
+ "tie_encoder_decoder": false,
61
+ "architectures": [
62
+ "Qwen2ForCausalLM"
63
+ ],
64
+ "finetuning_task": null,
65
+ "id2label": {
66
+ "0": "LABEL_0",
67
+ "1": "LABEL_1"
68
+ },
69
+ "label2id": {
70
+ "LABEL_0": 0,
71
+ "LABEL_1": 1
72
+ },
73
+ "task_specific_params": null,
74
+ "problem_type": null,
75
+ "tokenizer_class": null,
76
+ "prefix": null,
77
+ "bos_token_id": 151643,
78
+ "pad_token_id": null,
79
+ "eos_token_id": 151643,
80
+ "sep_token_id": null,
81
+ "decoder_start_token_id": null,
82
+ "max_length": 20,
83
+ "min_length": 0,
84
+ "do_sample": false,
85
+ "early_stopping": false,
86
+ "num_beams": 1,
87
+ "temperature": 1.0,
88
+ "top_k": 50,
89
+ "top_p": 1.0,
90
+ "typical_p": 1.0,
91
+ "repetition_penalty": 1.0,
92
+ "length_penalty": 1.0,
93
+ "no_repeat_ngram_size": 0,
94
+ "encoder_no_repeat_ngram_size": 0,
95
+ "bad_words_ids": null,
96
+ "num_return_sequences": 1,
97
+ "output_scores": false,
98
+ "return_dict_in_generate": false,
99
+ "forced_bos_token_id": null,
100
+ "forced_eos_token_id": null,
101
+ "remove_invalid_values": false,
102
+ "exponential_decay_length_penalty": null,
103
+ "suppress_tokens": null,
104
+ "begin_suppress_tokens": null,
105
+ "num_beam_groups": 1,
106
+ "diversity_penalty": 0.0,
107
+ "_name_or_path": "",
108
+ "transformers_version": "4.57.0",
109
+ "model_type": "qwen2",
110
+ "use_mrope": false,
111
+ "tf_legacy_loss": false,
112
+ "use_bfloat16": false,
113
+ "output_attentions": false
114
+ }
soar/merges.txt ADDED
The diff for this file is too large to render. See raw diff
 
soar/model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2787e6d4fe0b27ac33d28072abcadef53802c841bff037f3ee1f6bb5e1d3a2ce
3
+ size 4396289197
soar/speaker_encoder.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1cf3861c9dee79e4db34bd0b8a4155e68bed27a7c6274e168bb6ee4fed191c85
3
+ size 29150484
soar/special_tokens_map.json ADDED
@@ -0,0 +1,67 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "additional_special_tokens": [
3
+ {
4
+ "content": "<|audio_comp_start|>",
5
+ "lstrip": false,
6
+ "normalized": false,
7
+ "rstrip": false,
8
+ "single_word": false
9
+ },
10
+ {
11
+ "content": "<|audio_comp_span|>",
12
+ "lstrip": false,
13
+ "normalized": false,
14
+ "rstrip": false,
15
+ "single_word": false
16
+ },
17
+ {
18
+ "content": "<|audio_comp_end|>",
19
+ "lstrip": false,
20
+ "normalized": false,
21
+ "rstrip": false,
22
+ "single_word": false
23
+ },
24
+ {
25
+ "content": "<|audio_gen_start|>",
26
+ "lstrip": false,
27
+ "normalized": false,
28
+ "rstrip": false,
29
+ "single_word": false
30
+ },
31
+ {
32
+ "content": "<|audio_gen_span|>",
33
+ "lstrip": false,
34
+ "normalized": false,
35
+ "rstrip": false,
36
+ "single_word": false
37
+ },
38
+ {
39
+ "content": "<|audio_gen_end|>",
40
+ "lstrip": false,
41
+ "normalized": false,
42
+ "rstrip": false,
43
+ "single_word": false
44
+ },
45
+ {
46
+ "content": "<|text_cond_end|>",
47
+ "lstrip": false,
48
+ "normalized": false,
49
+ "rstrip": false,
50
+ "single_word": false
51
+ }
52
+ ],
53
+ "eos_token": {
54
+ "content": "<|endoftext|>",
55
+ "lstrip": false,
56
+ "normalized": false,
57
+ "rstrip": false,
58
+ "single_word": false
59
+ },
60
+ "pad_token": {
61
+ "content": "<|endoftext|>",
62
+ "lstrip": false,
63
+ "normalized": false,
64
+ "rstrip": false,
65
+ "single_word": false
66
+ }
67
+ }
soar/tokenizer.json ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c16521f66774c7a4774e5303b7c8ec5c99830c0be5aef6c6edde3ca2a5e05dd0
3
+ size 11423263
soar/tokenizer_config.json ADDED
@@ -0,0 +1,257 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "add_bos_token": false,
3
+ "add_prefix_space": false,
4
+ "added_tokens_decoder": {
5
+ "151643": {
6
+ "content": "<|endoftext|>",
7
+ "lstrip": false,
8
+ "normalized": false,
9
+ "rstrip": false,
10
+ "single_word": false,
11
+ "special": true
12
+ },
13
+ "151644": {
14
+ "content": "<|im_start|>",
15
+ "lstrip": false,
16
+ "normalized": false,
17
+ "rstrip": false,
18
+ "single_word": false,
19
+ "special": true
20
+ },
21
+ "151645": {
22
+ "content": "<|im_end|>",
23
+ "lstrip": false,
24
+ "normalized": false,
25
+ "rstrip": false,
26
+ "single_word": false,
27
+ "special": true
28
+ },
29
+ "151646": {
30
+ "content": "<|object_ref_start|>",
31
+ "lstrip": false,
32
+ "normalized": false,
33
+ "rstrip": false,
34
+ "single_word": false,
35
+ "special": true
36
+ },
37
+ "151647": {
38
+ "content": "<|object_ref_end|>",
39
+ "lstrip": false,
40
+ "normalized": false,
41
+ "rstrip": false,
42
+ "single_word": false,
43
+ "special": true
44
+ },
45
+ "151648": {
46
+ "content": "<|box_start|>",
47
+ "lstrip": false,
48
+ "normalized": false,
49
+ "rstrip": false,
50
+ "single_word": false,
51
+ "special": true
52
+ },
53
+ "151649": {
54
+ "content": "<|box_end|>",
55
+ "lstrip": false,
56
+ "normalized": false,
57
+ "rstrip": false,
58
+ "single_word": false,
59
+ "special": true
60
+ },
61
+ "151650": {
62
+ "content": "<|quad_start|>",
63
+ "lstrip": false,
64
+ "normalized": false,
65
+ "rstrip": false,
66
+ "single_word": false,
67
+ "special": true
68
+ },
69
+ "151651": {
70
+ "content": "<|quad_end|>",
71
+ "lstrip": false,
72
+ "normalized": false,
73
+ "rstrip": false,
74
+ "single_word": false,
75
+ "special": true
76
+ },
77
+ "151652": {
78
+ "content": "<|vision_start|>",
79
+ "lstrip": false,
80
+ "normalized": false,
81
+ "rstrip": false,
82
+ "single_word": false,
83
+ "special": true
84
+ },
85
+ "151653": {
86
+ "content": "<|vision_end|>",
87
+ "lstrip": false,
88
+ "normalized": false,
89
+ "rstrip": false,
90
+ "single_word": false,
91
+ "special": true
92
+ },
93
+ "151654": {
94
+ "content": "<|vision_pad|>",
95
+ "lstrip": false,
96
+ "normalized": false,
97
+ "rstrip": false,
98
+ "single_word": false,
99
+ "special": true
100
+ },
101
+ "151655": {
102
+ "content": "<|image_pad|>",
103
+ "lstrip": false,
104
+ "normalized": false,
105
+ "rstrip": false,
106
+ "single_word": false,
107
+ "special": true
108
+ },
109
+ "151656": {
110
+ "content": "<|video_pad|>",
111
+ "lstrip": false,
112
+ "normalized": false,
113
+ "rstrip": false,
114
+ "single_word": false,
115
+ "special": true
116
+ },
117
+ "151657": {
118
+ "content": "<tool_call>",
119
+ "lstrip": false,
120
+ "normalized": false,
121
+ "rstrip": false,
122
+ "single_word": false,
123
+ "special": false
124
+ },
125
+ "151658": {
126
+ "content": "</tool_call>",
127
+ "lstrip": false,
128
+ "normalized": false,
129
+ "rstrip": false,
130
+ "single_word": false,
131
+ "special": false
132
+ },
133
+ "151659": {
134
+ "content": "<|fim_prefix|>",
135
+ "lstrip": false,
136
+ "normalized": false,
137
+ "rstrip": false,
138
+ "single_word": false,
139
+ "special": false
140
+ },
141
+ "151660": {
142
+ "content": "<|fim_middle|>",
143
+ "lstrip": false,
144
+ "normalized": false,
145
+ "rstrip": false,
146
+ "single_word": false,
147
+ "special": false
148
+ },
149
+ "151661": {
150
+ "content": "<|fim_suffix|>",
151
+ "lstrip": false,
152
+ "normalized": false,
153
+ "rstrip": false,
154
+ "single_word": false,
155
+ "special": false
156
+ },
157
+ "151662": {
158
+ "content": "<|fim_pad|>",
159
+ "lstrip": false,
160
+ "normalized": false,
161
+ "rstrip": false,
162
+ "single_word": false,
163
+ "special": false
164
+ },
165
+ "151663": {
166
+ "content": "<|repo_name|>",
167
+ "lstrip": false,
168
+ "normalized": false,
169
+ "rstrip": false,
170
+ "single_word": false,
171
+ "special": false
172
+ },
173
+ "151664": {
174
+ "content": "<|file_sep|>",
175
+ "lstrip": false,
176
+ "normalized": false,
177
+ "rstrip": false,
178
+ "single_word": false,
179
+ "special": false
180
+ },
181
+ "151665": {
182
+ "content": "<|audio_comp_start|>",
183
+ "lstrip": false,
184
+ "normalized": false,
185
+ "rstrip": false,
186
+ "single_word": false,
187
+ "special": true
188
+ },
189
+ "151666": {
190
+ "content": "<|audio_comp_span|>",
191
+ "lstrip": false,
192
+ "normalized": false,
193
+ "rstrip": false,
194
+ "single_word": false,
195
+ "special": true
196
+ },
197
+ "151667": {
198
+ "content": "<|audio_comp_end|>",
199
+ "lstrip": false,
200
+ "normalized": false,
201
+ "rstrip": false,
202
+ "single_word": false,
203
+ "special": true
204
+ },
205
+ "151668": {
206
+ "content": "<|audio_gen_start|>",
207
+ "lstrip": false,
208
+ "normalized": false,
209
+ "rstrip": false,
210
+ "single_word": false,
211
+ "special": true
212
+ },
213
+ "151669": {
214
+ "content": "<|audio_gen_span|>",
215
+ "lstrip": false,
216
+ "normalized": false,
217
+ "rstrip": false,
218
+ "single_word": false,
219
+ "special": true
220
+ },
221
+ "151670": {
222
+ "content": "<|audio_gen_end|>",
223
+ "lstrip": false,
224
+ "normalized": false,
225
+ "rstrip": false,
226
+ "single_word": false,
227
+ "special": true
228
+ },
229
+ "151671": {
230
+ "content": "<|text_cond_end|>",
231
+ "lstrip": false,
232
+ "normalized": false,
233
+ "rstrip": false,
234
+ "single_word": false,
235
+ "special": true
236
+ }
237
+ },
238
+ "additional_special_tokens": [
239
+ "<|audio_comp_start|>",
240
+ "<|audio_comp_span|>",
241
+ "<|audio_comp_end|>",
242
+ "<|audio_gen_start|>",
243
+ "<|audio_gen_span|>",
244
+ "<|audio_gen_end|>",
245
+ "<|text_cond_end|>"
246
+ ],
247
+ "bos_token": null,
248
+ "clean_up_tokenization_spaces": false,
249
+ "eos_token": "<|endoftext|>",
250
+ "errors": "replace",
251
+ "extra_special_tokens": {},
252
+ "model_max_length": 131072,
253
+ "pad_token": "<|endoftext|>",
254
+ "split_special_tokens": false,
255
+ "tokenizer_class": "Qwen2Tokenizer",
256
+ "unk_token": null
257
+ }
soar/vocab.json ADDED
The diff for this file is too large to render. See raw diff
 
soar/vocoder.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c0e45c08f480df67ac4c354b465355fcc7e2f6c8765263b6dfeddd1f4671c93d
3
+ size 723585584