JinchuanTian commited on
Commit
df2e49a
·
verified ·
1 Parent(s): beb5682

Add files using upload-large-folder tool

Browse files
Files changed (11) hide show
  1. .gitattributes +0 -34
  2. LICENSE +201 -0
  3. MANIFEST.json +28 -0
  4. README.md +128 -0
  5. SHA256SUMS +9 -0
  6. THIRD_PARTY_NOTICES.md +40 -0
  7. deepspeed.json +45 -0
  8. inference_audio.yaml +17 -0
  9. inference_text.yaml +17 -0
  10. model.pt +3 -0
  11. train.yaml +45 -0
.gitattributes CHANGED
@@ -1,35 +1 @@
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
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
  *.pt filter=lfs diff=lfs merge=lfs -text
 
 
 
 
 
 
 
 
 
 
 
 
 
LICENSE ADDED
@@ -0,0 +1,201 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ Apache License
2
+ Version 2.0, January 2004
3
+ http://www.apache.org/licenses/
4
+
5
+ TERMS AND CONDITIONS FOR USE, REPRODUCTION, AND DISTRIBUTION
6
+
7
+ 1. Definitions.
8
+
9
+ "License" shall mean the terms and conditions for use, reproduction,
10
+ and distribution as defined by Sections 1 through 9 of this document.
11
+
12
+ "Licensor" shall mean the copyright owner or entity authorized by
13
+ the copyright owner that is granting the License.
14
+
15
+ "Legal Entity" shall mean the union of the acting entity and all
16
+ other entities that control, are controlled by, or are under common
17
+ control with that entity. For the purposes of this definition,
18
+ "control" means (i) the power, direct or indirect, to cause the
19
+ direction or management of such entity, whether by contract or
20
+ otherwise, or (ii) ownership of fifty percent (50%) or more of the
21
+ outstanding shares, or (iii) beneficial ownership of such entity.
22
+
23
+ "You" (or "Your") shall mean an individual or Legal Entity
24
+ exercising permissions granted by this License.
25
+
26
+ "Source" form shall mean the preferred form for making modifications,
27
+ including but not limited to software source code, documentation
28
+ source, and configuration files.
29
+
30
+ "Object" form shall mean any form resulting from mechanical
31
+ transformation or translation of a Source form, including but
32
+ not limited to compiled object code, generated documentation,
33
+ and conversions to other media types.
34
+
35
+ "Work" shall mean the work of authorship, whether in Source or
36
+ Object form, made available under the License, as indicated by a
37
+ copyright notice that is included in or attached to the work
38
+ (an example is provided in the Appendix below).
39
+
40
+ "Derivative Works" shall mean any work, whether in Source or Object
41
+ form, that is based on (or derived from) the Work and for which the
42
+ editorial revisions, annotations, elaborations, or other modifications
43
+ represent, as a whole, an original work of authorship. For the purposes
44
+ of this License, Derivative Works shall not include works that remain
45
+ separable from, or merely link (or bind by name) to the interfaces of,
46
+ the Work and Derivative Works thereof.
47
+
48
+ "Contribution" shall mean any work of authorship, including
49
+ the original version of the Work and any modifications or additions
50
+ to that Work or Derivative Works thereof, that is intentionally
51
+ submitted to Licensor for inclusion in the Work by the copyright owner
52
+ or by an individual or Legal Entity authorized to submit on behalf of
53
+ the copyright owner. For the purposes of this definition, "submitted"
54
+ means any form of electronic, verbal, or written communication sent
55
+ to the Licensor or its representatives, including but not limited to
56
+ communication on electronic mailing lists, source code control systems,
57
+ and issue tracking systems that are managed by, or on behalf of, the
58
+ Licensor for the purpose of discussing and improving the Work, but
59
+ excluding communication that is conspicuously marked or otherwise
60
+ designated in writing by the copyright owner as "Not a Contribution."
61
+
62
+ "Contributor" shall mean Licensor and any individual or Legal Entity
63
+ on behalf of whom a Contribution has been received by Licensor and
64
+ subsequently incorporated within the Work.
65
+
66
+ 2. Grant of Copyright License. Subject to the terms and conditions of
67
+ this License, each Contributor hereby grants to You a perpetual,
68
+ worldwide, non-exclusive, no-charge, royalty-free, irrevocable
69
+ copyright license to reproduce, prepare Derivative Works of,
70
+ publicly display, publicly perform, sublicense, and distribute the
71
+ Work and such Derivative Works in Source or Object form.
72
+
73
+ 3. Grant of Patent License. Subject to the terms and conditions of
74
+ this License, each Contributor hereby grants to You a perpetual,
75
+ worldwide, non-exclusive, no-charge, royalty-free, irrevocable
76
+ (except as stated in this section) patent license to make, have made,
77
+ use, offer to sell, sell, import, and otherwise transfer the Work,
78
+ where such license applies only to those patent claims licensable
79
+ by such Contributor that are necessarily infringed by their
80
+ Contribution(s) alone or by combination of their Contribution(s)
81
+ with the Work to which such Contribution(s) was submitted. If You
82
+ institute patent litigation against any entity (including a
83
+ cross-claim or counterclaim in a lawsuit) alleging that the Work
84
+ or a Contribution incorporated within the Work constitutes direct
85
+ or contributory patent infringement, then any patent licenses
86
+ granted to You under this License for that Work shall terminate
87
+ as of the date such litigation is filed.
88
+
89
+ 4. Redistribution. You may reproduce and distribute copies of the
90
+ Work or Derivative Works thereof in any medium, with or without
91
+ modifications, and in Source or Object form, provided that You
92
+ meet the following conditions:
93
+
94
+ (a) You must give any other recipients of the Work or
95
+ Derivative Works a copy of this License; and
96
+
97
+ (b) You must cause any modified files to carry prominent notices
98
+ stating that You changed the files; and
99
+
100
+ (c) You must retain, in the Source form of any Derivative Works
101
+ that You distribute, all copyright, patent, trademark, and
102
+ attribution notices from the Source form of the Work,
103
+ excluding those notices that do not pertain to any part of
104
+ the Derivative Works; and
105
+
106
+ (d) If the Work includes a "NOTICE" text file as part of its
107
+ distribution, then any Derivative Works that You distribute must
108
+ include a readable copy of the attribution notices contained
109
+ within such NOTICE file, excluding those notices that do not
110
+ pertain to any part of the Derivative Works, in at least one
111
+ of the following places: within a NOTICE text file distributed
112
+ as part of the Derivative Works; within the Source form or
113
+ documentation, if provided along with the Derivative Works; or,
114
+ within a display generated by the Derivative Works, if and
115
+ wherever such third-party notices normally appear. The contents
116
+ of the NOTICE file are for informational purposes only and
117
+ do not modify the License. You may add Your own attribution
118
+ notices within Derivative Works that You distribute, alongside
119
+ or as an addendum to the NOTICE text from the Work, provided
120
+ that such additional attribution notices cannot be construed
121
+ as modifying the License.
122
+
123
+ You may add Your own copyright statement to Your modifications and
124
+ may provide additional or different license terms and conditions
125
+ for use, reproduction, or distribution of Your modifications, or
126
+ for any such Derivative Works as a whole, provided Your use,
127
+ reproduction, and distribution of the Work otherwise complies with
128
+ the conditions stated in this License.
129
+
130
+ 5. Submission of Contributions. Unless You explicitly state otherwise,
131
+ any Contribution intentionally submitted for inclusion in the Work
132
+ by You to the Licensor shall be under the terms and conditions of
133
+ this License, without any additional terms or conditions.
134
+ Notwithstanding the above, nothing herein shall supersede or modify
135
+ the terms of any separate license agreement you may have executed
136
+ with Licensor regarding such Contributions.
137
+
138
+ 6. Trademarks. This License does not grant permission to use the trade
139
+ names, trademarks, service marks, or product names of the Licensor,
140
+ except as required for reasonable and customary use in describing the
141
+ origin of the Work and reproducing the content of the NOTICE file.
142
+
143
+ 7. Disclaimer of Warranty. Unless required by applicable law or
144
+ agreed to in writing, Licensor provides the Work (and each
145
+ Contributor provides its Contributions) on an "AS IS" BASIS,
146
+ WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or
147
+ implied, including, without limitation, any warranties or conditions
148
+ of TITLE, NON-INFRINGEMENT, MERCHANTABILITY, or FITNESS FOR A
149
+ PARTICULAR PURPOSE. You are solely responsible for determining the
150
+ appropriateness of using or redistributing the Work and assume any
151
+ risks associated with Your exercise of permissions under this License.
152
+
153
+ 8. Limitation of Liability. In no event and under no legal theory,
154
+ whether in tort (including negligence), contract, or otherwise,
155
+ unless required by applicable law (such as deliberate and grossly
156
+ negligent acts) or agreed to in writing, shall any Contributor be
157
+ liable to You for damages, including any direct, indirect, special,
158
+ incidental, or consequential damages of any character arising as a
159
+ result of this License or out of the use or inability to use the
160
+ Work (including but not limited to damages for loss of goodwill,
161
+ work stoppage, computer failure or malfunction, or any and all
162
+ other commercial damages or losses), even if such Contributor
163
+ has been advised of the possibility of such damages.
164
+
165
+ 9. Accepting Warranty or Additional Liability. While redistributing
166
+ the Work or Derivative Works thereof, You may choose to offer,
167
+ and charge a fee for, acceptance of support, warranty, indemnity,
168
+ or other liability obligations and/or rights consistent with this
169
+ License. However, in accepting such obligations, You may act only
170
+ on Your own behalf and on Your sole responsibility, not on behalf
171
+ of any other Contributor, and only if You agree to indemnify,
172
+ defend, and hold each Contributor harmless for any liability
173
+ incurred by, or claims asserted against, such Contributor by reason
174
+ of your accepting any such warranty or additional liability.
175
+
176
+ END OF TERMS AND CONDITIONS
177
+
178
+ APPENDIX: How to apply the Apache License to your work.
179
+
180
+ To apply the Apache License to your work, attach the following
181
+ boilerplate notice, with the fields enclosed by brackets "[]"
182
+ replaced with your own identifying information. (Don't include
183
+ the brackets!) The text should be enclosed in the appropriate
184
+ comment syntax for the file format. We also recommend that a
185
+ file or class name and description of purpose be included on the
186
+ same "printed page" as the copyright notice for easier
187
+ identification within third-party archives.
188
+
189
+ Copyright 2017 Johns Hopkins University (Shinji Watanabe)
190
+
191
+ Licensed under the Apache License, Version 2.0 (the "License");
192
+ you may not use this file except in compliance with the License.
193
+ You may obtain a copy of the License at
194
+
195
+ http://www.apache.org/licenses/LICENSE-2.0
196
+
197
+ Unless required by applicable law or agreed to in writing, software
198
+ distributed under the License is distributed on an "AS IS" BASIS,
199
+ WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
200
+ See the License for the specific language governing permissions and
201
+ limitations under the License.
MANIFEST.json ADDED
@@ -0,0 +1,28 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "artifact": "espnet/bagpiper-sft",
3
+ "format": "native ESPnet weight-only checkpoint",
4
+ "weight_file": {
5
+ "path": "model.pt",
6
+ "bytes": 18188823609,
7
+ "sha256": "50980756fad38bbefeb326ac0012e39f431ec5af8c0caaac8bd254f179f33746",
8
+ "top_level_keys": ["module"],
9
+ "tensor_count": 1382
10
+ },
11
+ "source": {
12
+ "repo": "JinchuanTian/bagpiper_sft",
13
+ "repo_type": "dataset",
14
+ "revision": "b11d5a0c11ad488edd04e3734d4bdff764977f57",
15
+ "path": "exp/opuslm_v2_stage2_pretrain_base/checkpoints/step_270000/global_step269985/mp_rank_00_model_states.pt",
16
+ "bytes": 18188293839,
17
+ "sha256": "e0d8fdd57c8cb8c938cd61ada43b20e469d9cea33ad8450e0971f80d4f9667b6"
18
+ },
19
+ "validation": {
20
+ "source_tensors_equal": 1381,
21
+ "added_tensor": "vocab_weight",
22
+ "strict_load": "pass",
23
+ "espnet_revision": "1992821bfb6d68c6636d1b75340b8b2b591bc015",
24
+ "transformers": "5.5.4",
25
+ "pytorch": "2.12.1+cu129",
26
+ "gpu": "NVIDIA H100 80GB HBM3"
27
+ }
28
+ }
README.md ADDED
@@ -0,0 +1,128 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ library_name: espnet
3
+ pipeline_tag: any-to-any
4
+ base_model: espnet/bagpiper
5
+ language:
6
+ - en
7
+ tags:
8
+ - audio
9
+ - multimodal
10
+ - speech-language-model
11
+ - audio-understanding
12
+ - audio-generation
13
+ - text-to-audio
14
+ - espnet
15
+ - pytorch
16
+ inference: false
17
+ ---
18
+
19
+ # Bagpiper SFT
20
+
21
+ Bagpiper is an 8B audio foundation model for open-ended audio understanding and
22
+ generation through rich captions. It accepts text and audio and can return text
23
+ or generate speech, music, environmental sounds, and mixtures of them.
24
+
25
+ This repository contains the paper-selected general-SFT checkpoint from
26
+ [Bagpiper: Solving Open-Ended Audio Tasks via Rich Captions](https://arxiv.org/abs/2602.05220).
27
+
28
+ ## Files and format
29
+
30
+ - `model.pt`: native ESPnet weight-only checkpoint, stored as
31
+ `{"module": state_dict}`.
32
+ - `train.yaml`: checkpoint-compatible model configuration.
33
+ - `inference_text.yaml`: audio-understanding/text-output decoding.
34
+ - `inference_audio.yaml`: text-then-audio generation decoding.
35
+ - `deepspeed.json`: optional fine-tuning configuration.
36
+ - `MANIFEST.json` and `SHA256SUMS`: provenance and integrity records.
37
+
38
+ `model.pt` contains 1,382 tensors: the 1,381 BF16 tensors from training plus the
39
+ deterministically reconstructed FP32 `vocab_weight` buffer required by the current
40
+ strict ESPnet loader. It contains no optimizer or exact-resume state. It is not a
41
+ Transformers `from_pretrained` directory and no vLLM compatibility is claimed.
42
+
43
+ ## Download
44
+
45
+ ```bash
46
+ hf download espnet/bagpiper-sft --local-dir bagpiper-sft
47
+ sha256sum -c bagpiper-sft/SHA256SUMS
48
+ ```
49
+
50
+ ## ESPnet runtime
51
+
52
+ The release was strict-loaded with:
53
+
54
+ - ESPnet PR [#6503](https://github.com/espnet/espnet/pull/6503), commit
55
+ `1992821bfb6d68c6636d1b75340b8b2b591bc015`;
56
+ - Python 3.11.15, PyTorch 2.12.1+cu129, Transformers 5.5.4;
57
+ - FlashAttention-3 3.0.0 at source commit
58
+ `77aacb68d194ba9af1010eda5eac3e7c0df8e6f6`; and
59
+ - one NVIDIA H100 80GB HBM3.
60
+
61
+ ```bash
62
+ git clone --branch titan-pt-weightonly-init https://github.com/jctian98/espnet.git
63
+ cd espnet
64
+ git checkout 1992821bfb6d68c6636d1b75340b8b2b591bc015
65
+ python -m pip install -e '.[speechlm]'
66
+ python -m pip install --no-build-isolation \
67
+ 'flash-attn-3 @ git+https://github.com/Dao-AILab/flash-attention.git@77aacb68d194ba9af1010eda5eac3e7c0df8e6f6#subdirectory=hopper'
68
+ ```
69
+
70
+ Use `espnet2/speechlm/bin/inference.py` with `train.yaml`, one of the supplied
71
+ inference YAMLs, and `model.pt`. The checkpoint can also initialize compatible
72
+ fine-tuning through `espnet2/speechlm/bin/train.py --resume-path model.pt`; this
73
+ starts a fresh optimizer, scheduler, and step counter.
74
+
75
+ ## Paper-reported results
76
+
77
+ These values are from the paper's checkpoint and evaluation protocols; they were
78
+ not rerun as part of the Hub upload.
79
+
80
+ | Evaluation | Result |
81
+ |---|---:|
82
+ | LibriSpeech test-clean WER | 2.5 |
83
+ | MMAU-Mini | 74.5 |
84
+ | MMAU | 73.1 |
85
+ | MMAR | 57.0 |
86
+ | AIR-Bench chat | 6.57 |
87
+ | AudioBench aggregate | 70.39 |
88
+ | General-model TTS WER | 2.7 |
89
+
90
+ The paper reports audio-fingerprint overlap between the SFT pool and **12.8% of
91
+ AudioBench** and **3.2% of AIR-Bench** evaluation clips. Interpret those results
92
+ with this disclosure.
93
+
94
+ ## Provenance and validation
95
+
96
+ - Selected checkpoint: `step_270000/global_step269985`.
97
+ - Raw source: `JinchuanTian/bagpiper_sft` at revision
98
+ `b11d5a0c11ad488edd04e3734d4bdff764977f57`.
99
+ - Raw SHA-256: `e0d8fdd57c8cb8c938cd61ada43b20e469d9cea33ad8450e0971f80d4f9667b6`.
100
+ - `model.pt` SHA-256:
101
+ `50980756fad38bbefeb326ac0012e39f431ec5af8c0caaac8bd254f179f33746`.
102
+ - All 1,381 source tensors matched the converted file exactly with
103
+ `torch.equal`; the only added key was `vocab_weight`.
104
+ - The 1,382-entry state dict passed the native ESPnet strict inference loader.
105
+
106
+ ## Limitations and responsible use
107
+
108
+ Understanding can hallucinate or mistranscribe. Generated audio can contain
109
+ artifacts, wrong words, unsafe content, or unintended resemblance to people and
110
+ styles. Most reported evaluations are English and some rely on model judges. The
111
+ model has no built-in watermark, factuality verifier, copyright filter, or
112
+ speaker-consent check. Do not use it for impersonation, deceptive media,
113
+ surveillance, biometric authentication, or high-stakes automated decisions.
114
+
115
+ Training-data availability and terms are documented at
116
+ [Bagpiper SFT Data](https://huggingface.co/datasets/espnet/Bagpiper_SFT_Data).
117
+ Third-party component terms are listed in `THIRD_PARTY_NOTICES.md`.
118
+
119
+ ## Citation
120
+
121
+ ```bibtex
122
+ @article{tian2026bagpiper,
123
+ title={Bagpiper: Solving Open-Ended Audio Tasks via Rich Captions},
124
+ author={Tian, Jinchuan and Wang, Haoran and Su, Bo-Hao and others},
125
+ journal={arXiv preprint arXiv:2602.05220},
126
+ year={2026}
127
+ }
128
+ ```
SHA256SUMS ADDED
@@ -0,0 +1,9 @@
 
 
 
 
 
 
 
 
 
 
1
+ 7f04d858ce14079bac4bc150278cbbd321f21512204953cc3e2db58a84a18fab README.md
2
+ 4696c3c9551da6fef1368be1e4ed2c80cf13e55448c6dcf2aba9462f5ff29ef5 LICENSE
3
+ 13e4484dd105cfc77a8fc41a1adc51869e381f70b6afeeb5469f5b75da80f3fb THIRD_PARTY_NOTICES.md
4
+ 7bf054db4d438bb66d4e3e6c0601b12bbed3e537d51305c8f7c4d9e2830604f0 MANIFEST.json
5
+ 38f04f7fc323963ee9cbff09167d60905e2c80d4d17a58fe7e5fa62b86321ae8 train.yaml
6
+ 3f8dddd19ba6de3d63bc5a82912955411b3b38c0a0880b21ddd2a0e210507212 deepspeed.json
7
+ 260c0dda42134194e480666d1f4cafb1d3ffd3b2ffeaf0af06ad9d1a21e3384a inference_audio.yaml
8
+ 66f59b0a0da7e5ecf67dd15075ee297572873de84836c834cce305c1cb176362 inference_text.yaml
9
+ 50980756fad38bbefeb326ac0012e39f431ec5af8c0caaac8bd254f179f33746 model.pt
THIRD_PARTY_NOTICES.md ADDED
@@ -0,0 +1,40 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # Third-party notices
2
+
3
+ `model.pt` contains parameters initialized from or associated with these components:
4
+
5
+ - Qwen3-8B-Base, revision `49e3418fbbbca6ecbdf9608b4d22e5a407081db4`, Apache-2.0:
6
+ https://huggingface.co/Qwen/Qwen3-8B-Base
7
+ - Qwen3-Omni-30B-A3B-Instruct audio encoder, revision
8
+ `26291f793822fb6be9555850f06dfe95f2d7e695`, Apache-2.0:
9
+ https://huggingface.co/Qwen/Qwen3-Omni-30B-A3B-Instruct
10
+ - X-Codec general-audio codec, revision
11
+ `240735a1ade315cb9b22604fe898e7df03d67378`, MIT:
12
+ https://huggingface.co/hf-audio/xcodec-hubert-general
13
+ - ESPnet SpeechLM code and release configuration, Apache-2.0:
14
+ https://github.com/espnet/espnet
15
+
16
+ Upstream names and trademarks do not imply endorsement.
17
+
18
+ ## X-Codec MIT notice
19
+
20
+ MIT License
21
+
22
+ Copyright (c) 2025 YE Zhen
23
+
24
+ Permission is hereby granted, free of charge, to any person obtaining a copy
25
+ of this software and associated documentation files (the "Software"), to deal
26
+ in the Software without restriction, including without limitation the rights
27
+ to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
28
+ copies of the Software, and to permit persons to whom the Software is
29
+ furnished to do so, subject to the following conditions:
30
+
31
+ The above copyright notice and this permission notice shall be included in all
32
+ copies or substantial portions of the Software.
33
+
34
+ THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
35
+ IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
36
+ FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
37
+ AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
38
+ LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
39
+ OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
40
+ SOFTWARE.
deepspeed.json ADDED
@@ -0,0 +1,45 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "train_micro_batch_size_per_gpu": 1,
3
+ "gradient_accumulation_steps": 1,
4
+ "gradient_clipping": 1.0,
5
+ "bf16": {"enabled": true},
6
+ "checkpoint": {"load_universal": true},
7
+ "optimizer": {
8
+ "type": "Adam",
9
+ "params": {
10
+ "lr": 0.00001,
11
+ "betas": [0.9, 0.95],
12
+ "eps": 1e-8,
13
+ "weight_decay": 3e-7,
14
+ "adam_w_mode": true
15
+ }
16
+ },
17
+ "zero_optimization": {
18
+ "stage": 1,
19
+ "allgather_partitions": true,
20
+ "allgather_bucket_size": 300000000,
21
+ "reduce_scatter": true,
22
+ "reduce_bucket_size": 300000000,
23
+ "overlap_comm": true,
24
+ "contiguous_gradients": true
25
+ },
26
+ "activation_checkpointing": {
27
+ "partition_activations": false,
28
+ "cpu_checkpointing": false,
29
+ "contiguous_memory_optimization": true,
30
+ "number_checkpoints": null,
31
+ "synchronize_checkpoint_boundary": false,
32
+ "profile": false
33
+ },
34
+ "scheduler": {
35
+ "type": "WarmupCosineLR",
36
+ "params": {
37
+ "warmup_type": "linear",
38
+ "total_num_steps": 600000,
39
+ "warmup_num_steps": 350000,
40
+ "cos_min_ratio": 1.0
41
+ }
42
+ },
43
+ "wall_clock_breakdown": false,
44
+ "steps_per_print": 1000
45
+ }
inference_audio.yaml ADDED
@@ -0,0 +1,17 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ dtype: bfloat16
2
+ num_hypo: 1
3
+ enforce_modality: ["text", "audio"]
4
+
5
+ audio:
6
+ temperature: 0.8
7
+ topk: 20
8
+ cfg: 3
9
+ max_step: 2048
10
+ min_step: 50
11
+
12
+ text:
13
+ temperature: 0.6
14
+ topk: 20
15
+ cfg: 1
16
+ max_step: 2048
17
+ min_step: 1
inference_text.yaml ADDED
@@ -0,0 +1,17 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ dtype: bfloat16
2
+ num_hypo: 1
3
+ enforce_modality: ["text"]
4
+
5
+ audio:
6
+ temperature: 0.8
7
+ topk: 20
8
+ cfg: 3
9
+ max_step: 2048
10
+ min_step: 50
11
+
12
+ text:
13
+ temperature: 0.6
14
+ topk: 20
15
+ cfg: 1
16
+ max_step: 2048
17
+ min_step: 1
model.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:50980756fad38bbefeb326ac0012e39f431ec5af8c0caaac8bd254f179f33746
3
+ size 18188823609
train.yaml ADDED
@@ -0,0 +1,45 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ job_type: speechlm
2
+
3
+ multimodal_io:
4
+ text:
5
+ tokenizer_name: Qwen/Qwen3-8B-Base
6
+ discrete_audio:
7
+ codec_choice: Xcodec
8
+ codec_hf_model_tag: hf-audio/xcodec-hubert-general
9
+ ssl_choice: null
10
+ ssl_hf_model_tag: null
11
+ delay_interleave: true
12
+ stream_weights: [0.125, 0.125, 0.125, 0.125, 0.125, 0.125, 0.125, 0.125]
13
+ continuous_audio:
14
+ encoder_choice: huggingface
15
+ encoder_hf_model_tag: Qwen/Qwen3-Omni-30B-A3B-Instruct
16
+ attn_implementation: flash_attention_3
17
+ dtype: bfloat16
18
+
19
+ model:
20
+ model_choice: parallel
21
+ model_hf_tag: Qwen/Qwen3-8B-Base
22
+ model_conf:
23
+ attn_implementation: flash_attention_3
24
+ dtype: bfloat16
25
+ activation_checkpointing: true
26
+
27
+ preprocessor:
28
+ audio_input: continuous_audio
29
+ audio_output: discrete_audio
30
+ loss_region: assistant
31
+ audio_cfg: 0.05
32
+
33
+ data_loading:
34
+ batchfy_method: pack
35
+ batch_size: 20000
36
+ save_loader_state: false
37
+ seed: 43
38
+ num_workers: 6
39
+
40
+ trainer:
41
+ deepspeed_config: deepspeed.json
42
+ freeze_param: [multimodal_io_dict.discrete_audio, multimodal_io_dict.continuous_audio]
43
+ max_step: 700000
44
+ save_interval: 2500
45
+ log_interval: 1