Add AuK Base and AuK-Flash GGUF components
Browse files- .gitattributes +42 -0
- README.md +88 -0
- auk-base-f16.gguf +3 -0
- auk-base-f32.gguf +3 -0
- auk-base-q8_0.gguf +3 -0
- auk-flash-f16.gguf +3 -0
- auk-flash-f32.gguf +3 -0
- auk-flash-q8_0.gguf +3 -0
- auk-vae-f32.gguf +3 -0
- config/auk-base.yaml +45 -0
- config/auk-flash.yaml +45 -0
- examples/cpp/add_cough.wav +3 -0
- examples/cpp/deep_voice.wav +3 -0
- examples/cpp/enhance.wav +3 -0
- examples/cpp/get_what_speaker.wav +3 -0
- examples/cpp/happy.wav +3 -0
- examples/cpp/increase_volume.wav +3 -0
- examples/cpp/instruct.wav +3 -0
- examples/cpp/raise_pitch.wav +3 -0
- examples/cpp/replace_lyrics.wav +3 -0
- examples/cpp/replace_words.wav +3 -0
- examples/cpp/second_speaker.wav +3 -0
- examples/cpp/speed_1_5x.wav +3 -0
- examples/cpp/standard_mandarin.wav +3 -0
- examples/cpp/vocals_only.wav +3 -0
- examples/cpp/whisper.wav +3 -0
- examples/cpp/zero_shot.wav +3 -0
- examples/python/add_cough.wav +3 -0
- examples/python/deep_voice.wav +3 -0
- examples/python/enhance.wav +3 -0
- examples/python/get_what_speaker.wav +3 -0
- examples/python/happy.wav +3 -0
- examples/python/increase_volume.wav +3 -0
- examples/python/instruct.wav +3 -0
- examples/python/raise_pitch.wav +3 -0
- examples/python/replace_lyrics.wav +3 -0
- examples/python/replace_words.wav +3 -0
- examples/python/second_speaker.wav +3 -0
- examples/python/speed_1_5x.wav +3 -0
- examples/python/standard_mandarin.wav +3 -0
- examples/python/vocals_only.wav +3 -0
- examples/python/whisper.wav +3 -0
- examples/python/zero_shot.wav +3 -0
- qwen2.5-omni-3b-bf16.gguf +3 -0
- qwen2.5-omni-3b-q8_0.gguf +3 -0
- tokenizer/tokenizer.json +3 -0
- tokenizer/tokenizer_config.json +223 -0
.gitattributes
CHANGED
|
@@ -33,3 +33,45 @@ 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 |
+
auk-base-f16.gguf filter=lfs diff=lfs merge=lfs -text
|
| 37 |
+
auk-base-f32.gguf filter=lfs diff=lfs merge=lfs -text
|
| 38 |
+
auk-base-q8_0.gguf filter=lfs diff=lfs merge=lfs -text
|
| 39 |
+
auk-flash-f16.gguf filter=lfs diff=lfs merge=lfs -text
|
| 40 |
+
auk-flash-f32.gguf filter=lfs diff=lfs merge=lfs -text
|
| 41 |
+
auk-flash-q8_0.gguf filter=lfs diff=lfs merge=lfs -text
|
| 42 |
+
auk-vae-f32.gguf filter=lfs diff=lfs merge=lfs -text
|
| 43 |
+
examples/cpp/add_cough.wav filter=lfs diff=lfs merge=lfs -text
|
| 44 |
+
examples/cpp/deep_voice.wav filter=lfs diff=lfs merge=lfs -text
|
| 45 |
+
examples/cpp/enhance.wav filter=lfs diff=lfs merge=lfs -text
|
| 46 |
+
examples/cpp/get_what_speaker.wav filter=lfs diff=lfs merge=lfs -text
|
| 47 |
+
examples/cpp/happy.wav filter=lfs diff=lfs merge=lfs -text
|
| 48 |
+
examples/cpp/increase_volume.wav filter=lfs diff=lfs merge=lfs -text
|
| 49 |
+
examples/cpp/instruct.wav filter=lfs diff=lfs merge=lfs -text
|
| 50 |
+
examples/cpp/raise_pitch.wav filter=lfs diff=lfs merge=lfs -text
|
| 51 |
+
examples/cpp/replace_lyrics.wav filter=lfs diff=lfs merge=lfs -text
|
| 52 |
+
examples/cpp/replace_words.wav filter=lfs diff=lfs merge=lfs -text
|
| 53 |
+
examples/cpp/second_speaker.wav filter=lfs diff=lfs merge=lfs -text
|
| 54 |
+
examples/cpp/speed_1_5x.wav filter=lfs diff=lfs merge=lfs -text
|
| 55 |
+
examples/cpp/standard_mandarin.wav filter=lfs diff=lfs merge=lfs -text
|
| 56 |
+
examples/cpp/vocals_only.wav filter=lfs diff=lfs merge=lfs -text
|
| 57 |
+
examples/cpp/whisper.wav filter=lfs diff=lfs merge=lfs -text
|
| 58 |
+
examples/cpp/zero_shot.wav filter=lfs diff=lfs merge=lfs -text
|
| 59 |
+
examples/python/add_cough.wav filter=lfs diff=lfs merge=lfs -text
|
| 60 |
+
examples/python/deep_voice.wav filter=lfs diff=lfs merge=lfs -text
|
| 61 |
+
examples/python/enhance.wav filter=lfs diff=lfs merge=lfs -text
|
| 62 |
+
examples/python/get_what_speaker.wav filter=lfs diff=lfs merge=lfs -text
|
| 63 |
+
examples/python/happy.wav filter=lfs diff=lfs merge=lfs -text
|
| 64 |
+
examples/python/increase_volume.wav filter=lfs diff=lfs merge=lfs -text
|
| 65 |
+
examples/python/instruct.wav filter=lfs diff=lfs merge=lfs -text
|
| 66 |
+
examples/python/raise_pitch.wav filter=lfs diff=lfs merge=lfs -text
|
| 67 |
+
examples/python/replace_lyrics.wav filter=lfs diff=lfs merge=lfs -text
|
| 68 |
+
examples/python/replace_words.wav filter=lfs diff=lfs merge=lfs -text
|
| 69 |
+
examples/python/second_speaker.wav filter=lfs diff=lfs merge=lfs -text
|
| 70 |
+
examples/python/speed_1_5x.wav filter=lfs diff=lfs merge=lfs -text
|
| 71 |
+
examples/python/standard_mandarin.wav filter=lfs diff=lfs merge=lfs -text
|
| 72 |
+
examples/python/vocals_only.wav filter=lfs diff=lfs merge=lfs -text
|
| 73 |
+
examples/python/whisper.wav filter=lfs diff=lfs merge=lfs -text
|
| 74 |
+
examples/python/zero_shot.wav filter=lfs diff=lfs merge=lfs -text
|
| 75 |
+
qwen2.5-omni-3b-bf16.gguf filter=lfs diff=lfs merge=lfs -text
|
| 76 |
+
qwen2.5-omni-3b-q8_0.gguf filter=lfs diff=lfs merge=lfs -text
|
| 77 |
+
tokenizer/tokenizer.json filter=lfs diff=lfs merge=lfs -text
|
README.md
ADDED
|
@@ -0,0 +1,88 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
---
|
| 2 |
+
license: mit
|
| 3 |
+
base_model: tencent/AuK
|
| 4 |
+
pipeline_tag: text-to-speech
|
| 5 |
+
tags:
|
| 6 |
+
- audio
|
| 7 |
+
- speech
|
| 8 |
+
- text-to-speech
|
| 9 |
+
- zero-shot-tts
|
| 10 |
+
- voice-cloning
|
| 11 |
+
- speech-generation
|
| 12 |
+
- speech-editing
|
| 13 |
+
- speech-enhancement
|
| 14 |
+
- source-separation
|
| 15 |
+
- speech-separation
|
| 16 |
+
- instruction-guided
|
| 17 |
+
- diffusion
|
| 18 |
+
- audio.cpp
|
| 19 |
+
---
|
| 20 |
+
|
| 21 |
+
# AuK GGUF for audio.cpp
|
| 22 |
+
|
| 23 |
+
These GGUF files are for use with [audio.cpp](https://github.com/0xShug0/audio.cpp). The card's `base_model` points to the [Tencent AuK base model](https://huggingface.co/tencent/AuK). This package also includes optional files converted from [AuK-Flash](https://huggingface.co/tencent/AuK-Flash); Flash is a separate distilled variant and is not the pinned base model.
|
| 24 |
+
|
| 25 |
+
> **Output quality:** We validate parity with upstream Python for the tested configuration, not whether AuK's output meets every quality expectation. If a result is disappointing, listen to the corresponding Python reference WAV linked below first. A similar Python result points to the upstream model's behavior, not necessarily an audio.cpp conversion issue. The quantized component combinations have not received the same 16-task parity validation.
|
| 26 |
+
|
| 27 |
+
Pass the downloaded directory to `audiocpp_cli --model`. Keep the `config/` and `tokenizer/` directories alongside the GGUF files. The default components are `auk-base-f32.gguf`, `qwen2.5-omni-3b-bf16.gguf`, and `auk-vae-f32.gguf`. AuK-Flash requires `--session-option auk.variant=flash`; its default generator is `auk-flash-f32.gguf`. The Qwen component is Qwen2.5-Omni-3B.
|
| 28 |
+
|
| 29 |
+
For example, instruction TTS with explicitly selected Base components:
|
| 30 |
+
|
| 31 |
+
```bash
|
| 32 |
+
audiocpp_cli --task tts --family auk \
|
| 33 |
+
--model /path/to/AuK-Base-and-Flash-GGUF --backend cuda \
|
| 34 |
+
--session-option auk.variant=base \
|
| 35 |
+
--session-option auk.model_gguf=auk-base-f32.gguf \
|
| 36 |
+
--session-option auk.qwen_gguf=qwen2.5-omni-3b-bf16.gguf \
|
| 37 |
+
--session-option auk.vae_gguf=auk-vae-f32.gguf \
|
| 38 |
+
--text "The next train leaves in ten minutes." \
|
| 39 |
+
--request-option "instruct=A warm male voice speaking clear English." \
|
| 40 |
+
--request-option duration_sec=4 --seed 42 --log --out speech.wav
|
| 41 |
+
```
|
| 42 |
+
|
| 43 |
+
To use AuK-Flash, select `auk.variant=flash` and an `auk-flash-*.gguf` generator. To use the Qwen Q8_0 component, select `auk.qwen_gguf=qwen2.5-omni-3b-q8_0.gguf`. Keep the F32 VAE in either case.
|
| 44 |
+
|
| 45 |
+
For editing, use `--task gen --audio input.wav --text "<editing instruction>"`. See the [audio.cpp AuK guide](https://github.com/0xShug0/audio.cpp/blob/main/docs/models/auk.md) and the [AuK cookbook](https://github.com/Tencent-Hunyuan/AuK/blob/main/docs/COOKBOOK.md) for request options and example instructions.
|
| 46 |
+
|
| 47 |
+
## Sixteen-task validation
|
| 48 |
+
|
| 49 |
+
The 16-task C++/Python comparison forced FP32 inference and disabled TF32 in both implementations. Input audio and reference voices came from the [upstream AuK demo assets](https://github.com/Tencent-Hunyuan/AuK/tree/main/assets/demo-input-audio). Durations not shown below matched the source recording.
|
| 50 |
+
|
| 51 |
+
All 16 C++ requests completed and produced 24 kHz WAVs with the same frame counts as the Python outputs. The cosine values compare C++ WAVs with the Python baseline. Waveform cosine alone does not establish whether an edit followed its instruction.
|
| 52 |
+
|
| 53 |
+
| Task | Tested instruction / setting | C++ vs Python WAV cosine | Listen |
|
| 54 |
+
| --- | --- | ---: | --- |
|
| 55 |
+
| Zero-shot TTS | Reference voice; target sentence; 6 s | 0.999999992 | [Python](./examples/python/zero_shot.wav) / [C++](./examples/cpp/zero_shot.wav) |
|
| 56 |
+
| Instruct TTS | Calm woman speaking clear English; 3 s | 0.999999898 | [Python](./examples/python/instruct.wav) / [C++](./examples/cpp/instruct.wav) |
|
| 57 |
+
| Speech content editing | Replace "but accepting what we cannot have" with "and living well with dreams unmet"; 7 s | 0.999999982 | [Python](./examples/python/replace_words.wav) / [C++](./examples/cpp/replace_words.wav) |
|
| 58 |
+
| Lyric editing | Replace "rear view" with "like you" in isolated vocals | 1.000000000 | [Python](./examples/python/replace_lyrics.wav) / [C++](./examples/cpp/replace_lyrics.wav) |
|
| 59 |
+
| Pitch editing | Raise pitch by 2 semitones | 0.999989991 | [Python](./examples/python/raise_pitch.wav) / [C++](./examples/cpp/raise_pitch.wav) |
|
| 60 |
+
| Speed editing | Speed 1.5x; 6.86 s | 0.999999993 | [Python](./examples/python/speed_1_5x.wav) / [C++](./examples/cpp/speed_1_5x.wav) |
|
| 61 |
+
| Volume editing | Increase volume by 10 dB | 0.999999997 | [Python](./examples/python/increase_volume.wav) / [C++](./examples/cpp/increase_volume.wav) |
|
| 62 |
+
| Emotion editing | Change emotion to happy | 0.999999993 | [Python](./examples/python/happy.wav) / [C++](./examples/cpp/happy.wav) |
|
| 63 |
+
| Timbre editing | Change to a deep, calm male voice | 0.999999991 | [Python](./examples/python/deep_voice.wav) / [C++](./examples/cpp/deep_voice.wav) |
|
| 64 |
+
| De-accent | Change Sichuan-accented speech to standard Mandarin | 0.999999795 | [Python](./examples/python/standard_mandarin.wav) / [C++](./examples/cpp/standard_mandarin.wav) |
|
| 65 |
+
| Nonverbal editing | Add a cough before "We tested"; 10.44 s | 0.999991188 | [Python](./examples/python/add_cough.wav) / [C++](./examples/cpp/add_cough.wav) |
|
| 66 |
+
| Whisper conversion | Speak the source in a quiet whisper | 0.999998371 | [Python](./examples/python/whisper.wav) / [C++](./examples/cpp/whisper.wav) |
|
| 67 |
+
| Speech enhancement | Preserve speakers; remove noise and reverberation | 0.999999994 | [Python](./examples/python/enhance.wav) / [C++](./examples/cpp/enhance.wav) |
|
| 68 |
+
| Speech separation | Keep the second speaker to start talking | 0.999999985 | [Python](./examples/python/second_speaker.wav) / [C++](./examples/cpp/second_speaker.wav) |
|
| 69 |
+
| Music separation | Keep the singing voice; remove other audio | 0.999999998 | [Python](./examples/python/vocals_only.wav) / [C++](./examples/cpp/vocals_only.wav) |
|
| 70 |
+
| Target speaker extraction | Keep the speaker saying "get what" | 0.999999997 | [Python](./examples/python/get_what_speaker.wav) / [C++](./examples/cpp/get_what_speaker.wav) |
|
| 71 |
+
|
| 72 |
+
The no-TF32 C++ run matched the Python WAVs at waveform cosine 0.999989991 or higher for all 16 tasks; the files were not byte-identical. These historical results use the original combined GGUF and are **not** a 16-task parity claim for the component GGUFs in this package.
|
| 73 |
+
|
| 74 |
+
## Dtypes and validation scope
|
| 75 |
+
|
| 76 |
+
All component smoke tests used CUDA and the F32 VAE. A pass here means the listed task loaded the selected GGUFs and generated a WAV; it does not establish Python parity or task quality.
|
| 77 |
+
|
| 78 |
+
| Generator | Qwen conditioner | Tested task(s) |
|
| 79 |
+
| --- | --- | --- |
|
| 80 |
+
| Base F32 | BF16 | Zero-shot TTS; instruct TTS |
|
| 81 |
+
| Base F16 or Q8_0 | BF16 | Instruct TTS, one run per dtype |
|
| 82 |
+
| Base F32 or Q8_0 | Q8_0 | Zero-shot TTS, one run per combination |
|
| 83 |
+
| AuK-Flash F32 | BF16 | Zero-shot TTS; instruct TTS |
|
| 84 |
+
| AuK-Flash F16 or Q8_0 | BF16 | Instruct TTS, one run per dtype |
|
| 85 |
+
| AuK-Flash F32 | Q8_0 | Zero-shot TTS, one run |
|
| 86 |
+
| AuK-Flash F16 | Q8_0 | Instruct TTS, one run |
|
| 87 |
+
|
| 88 |
+
The Base F16 + Qwen Q8_0 and AuK-Flash Q8_0 + Qwen Q8_0 combinations were **not tested**. Full 16-task Python parity is untested for every component combination, including the default Base F32 and AuK-Flash F32 paths. The VAE is supplied only as F32; no F16 or quantized VAE is supplied or tested. CPU and Vulkan parity are also untested.
|
auk-base-f16.gguf
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:d7efc034ef18aa6d5080cb8a3a67731df31aeae7f73d68b19400503db89bd586
|
| 3 |
+
size 3061149120
|
auk-base-f32.gguf
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:a0ab5bffe8db0c8a9b54ec357ec8a4bd3d3d991a796859d4b9ad55e2b17af93b
|
| 3 |
+
size 6122226368
|
auk-base-q8_0.gguf
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:dcf96485cf4d44f8dd84eb190348aa3354117380c95e9b3d90b1f4c797503a39
|
| 3 |
+
size 1639358080
|
auk-flash-f16.gguf
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:f9ad537dbe1f207943a42ee48dad070fb91b62c7b812e0a124f16dfba72dc092
|
| 3 |
+
size 3061149120
|
auk-flash-f32.gguf
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:8c1592b6aa12eeb4155e060c21efa9caf681e7c0cfd1afc45a11907d69dc0c55
|
| 3 |
+
size 6122226368
|
auk-flash-q8_0.gguf
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:ec6aab06181307a4e7ba9e22d21532338d9f673162e8933c4cc2d9d6dac45e33
|
| 3 |
+
size 1639358080
|
auk-vae-f32.gguf
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:401e96baa1ac958d66fbe2c8cf5d25105c995c46dd46247c879e6114ef503abd
|
| 3 |
+
size 637364224
|
config/auk-base.yaml
ADDED
|
@@ -0,0 +1,45 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
model:
|
| 2 |
+
name: AuK
|
| 3 |
+
backbone: Flux2Edit
|
| 4 |
+
cfm_class: CFMEdit
|
| 5 |
+
|
| 6 |
+
schedule:
|
| 7 |
+
t_sampling: logistic_normal
|
| 8 |
+
P_mean: -0.8
|
| 9 |
+
P_std: 0.8
|
| 10 |
+
|
| 11 |
+
text_encoder:
|
| 12 |
+
text_encoder_path: ckpts/Qwen2.5-Omni-3B
|
| 13 |
+
|
| 14 |
+
arch:
|
| 15 |
+
dim: 1536
|
| 16 |
+
heads: 24
|
| 17 |
+
ff_mult: 2
|
| 18 |
+
text_hidden_dim: 2048
|
| 19 |
+
attn_backend: flash_attn # CLI overrides to "torch" for inference (no flash_attn dep)
|
| 20 |
+
attn_mask_enabled: True
|
| 21 |
+
checkpoint_activations: True # gradient checkpointing: peak ~91G→~75G at every_n=4
|
| 22 |
+
checkpoint_every_n_layers: 4 # checkpoint every 4th layer (fewer recomputes, faster than every=1)
|
| 23 |
+
num_layers: 10
|
| 24 |
+
num_single_layers: 20
|
| 25 |
+
|
| 26 |
+
vae:
|
| 27 |
+
vae_name: BigVGANFlowVAE
|
| 28 |
+
vae_model_path: ckpts/AuK/vae.safetensors
|
| 29 |
+
latent_dim: 64
|
| 30 |
+
downsample_rate: 480
|
| 31 |
+
target_sample_rate: 24000
|
| 32 |
+
model_init_kwargs:
|
| 33 |
+
upsample_rates: [5, 4, 3, 2, 2, 2]
|
| 34 |
+
upsample_kernel_sizes: [10, 8, 6, 4, 4, 4]
|
| 35 |
+
upsample_initial_channel: 1536
|
| 36 |
+
resblock_kernel_sizes: [3, 7, 11]
|
| 37 |
+
resblock_dilation_sizes: [[1, 3, 5], [1, 3, 5], [1, 3, 5]]
|
| 38 |
+
downsample_rates: [2, 2, 2, 3, 4, 5]
|
| 39 |
+
downsample_channels: [12, 24, 48, 96, 192, 384, 768]
|
| 40 |
+
snake_logscale: true
|
| 41 |
+
latent_dim: ${model.vae.latent_dim}
|
| 42 |
+
use_vae: true
|
| 43 |
+
causal: true
|
| 44 |
+
flow_hidden_channels: 256
|
| 45 |
+
act_causal: true
|
config/auk-flash.yaml
ADDED
|
@@ -0,0 +1,45 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
model:
|
| 2 |
+
name: AuK-Flash
|
| 3 |
+
backbone: Flux2Edit
|
| 4 |
+
cfm_class: CFMEdit
|
| 5 |
+
|
| 6 |
+
schedule:
|
| 7 |
+
t_sampling: logistic_normal
|
| 8 |
+
P_mean: -0.8
|
| 9 |
+
P_std: 0.8
|
| 10 |
+
|
| 11 |
+
text_encoder:
|
| 12 |
+
text_encoder_path: ckpts/Qwen2.5-Omni-3B
|
| 13 |
+
|
| 14 |
+
arch:
|
| 15 |
+
dim: 1536
|
| 16 |
+
heads: 24
|
| 17 |
+
ff_mult: 2
|
| 18 |
+
text_hidden_dim: 2048
|
| 19 |
+
attn_backend: flash_attn # CLI overrides to "torch" for inference (no flash_attn dep)
|
| 20 |
+
attn_mask_enabled: True
|
| 21 |
+
checkpoint_activations: True # gradient checkpointing: peak ~91G→~75G at every_n=4
|
| 22 |
+
checkpoint_every_n_layers: 4 # checkpoint every 4th layer (fewer recomputes, faster than every=1)
|
| 23 |
+
num_layers: 10
|
| 24 |
+
num_single_layers: 20
|
| 25 |
+
|
| 26 |
+
vae:
|
| 27 |
+
vae_name: BigVGANFlowVAE
|
| 28 |
+
vae_model_path: ckpts/AuK-Flash/vae.safetensors
|
| 29 |
+
latent_dim: 64
|
| 30 |
+
downsample_rate: 480
|
| 31 |
+
target_sample_rate: 24000
|
| 32 |
+
model_init_kwargs:
|
| 33 |
+
upsample_rates: [5, 4, 3, 2, 2, 2]
|
| 34 |
+
upsample_kernel_sizes: [10, 8, 6, 4, 4, 4]
|
| 35 |
+
upsample_initial_channel: 1536
|
| 36 |
+
resblock_kernel_sizes: [3, 7, 11]
|
| 37 |
+
resblock_dilation_sizes: [[1, 3, 5], [1, 3, 5], [1, 3, 5]]
|
| 38 |
+
downsample_rates: [2, 2, 2, 3, 4, 5]
|
| 39 |
+
downsample_channels: [12, 24, 48, 96, 192, 384, 768]
|
| 40 |
+
snake_logscale: true
|
| 41 |
+
latent_dim: ${model.vae.latent_dim}
|
| 42 |
+
use_vae: true
|
| 43 |
+
causal: true
|
| 44 |
+
flow_hidden_channels: 256
|
| 45 |
+
act_causal: true
|
examples/cpp/add_cough.wav
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:3a1ca0a5f7acf99c08b62aba389e31367ec24a03bf4414769e90a7e18825bed4
|
| 3 |
+
size 501164
|
examples/cpp/deep_voice.wav
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:18b434356c751d94709c29d7289d01e160025c975e0a6c75aa1f1d73fbf08a8a
|
| 3 |
+
size 316844
|
examples/cpp/enhance.wav
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:7f4e1d9bbf8b26ca4915ac897759041bfa85b8d964b3767c62236812fe7567df
|
| 3 |
+
size 239084
|
examples/cpp/get_what_speaker.wav
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:59f9ad9f3616130f3d5a9023afce399d815a0a9a1bf33bf46708d2420a7aeae6
|
| 3 |
+
size 1344044
|
examples/cpp/happy.wav
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:9489b4a2b959d1df45fac4124da5a6495a6bfe5bf463091be5783893669add3f
|
| 3 |
+
size 303404
|
examples/cpp/increase_volume.wav
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:6ea7cf7d76bde0214eedbe4e6faa4cfeb15c7afb484392add730985090d32287
|
| 3 |
+
size 264044
|
examples/cpp/instruct.wav
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:97bd692a509a5add0924961d4ce23d615002323b509a66e1a49c8485a2aa4c80
|
| 3 |
+
size 144044
|
examples/cpp/raise_pitch.wav
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:2cd2d7c90531bc652af4e1f6fe08b9fe078f70bbe8ac69d0a43ef57554e43b59
|
| 3 |
+
size 264044
|
examples/cpp/replace_lyrics.wav
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:b9a21df14b84c439030ee437c7c865f3b0d8aa98f91f36ab0f2e4a8fe64cc238
|
| 3 |
+
size 275564
|
examples/cpp/replace_words.wav
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:d82a964b77e2e13a4196e1897a79b0b4e6aa1ac687380424f40d198c2b7b7e9d
|
| 3 |
+
size 336044
|
examples/cpp/second_speaker.wav
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:172b3fb42ad3cb48c3cf535b9e8dfd01bfa61aea1051db4b93288716d5bd66fe
|
| 3 |
+
size 922604
|
examples/cpp/speed_1_5x.wav
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:07397bf90ae1db2bd2b8f451aabf711cdcdafc96d20a2454b9e7da052830f39b
|
| 3 |
+
size 329324
|
examples/cpp/standard_mandarin.wav
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:2de25af9e8a649ca51daca455826659b8648919f8e4a7c96bf7c9ce0e840efbb
|
| 3 |
+
size 393644
|
examples/cpp/vocals_only.wav
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:95f32d7823a13a5346e46c7a84c97e4d6b5497ab34c4c8ad3a82bfc551873d85
|
| 3 |
+
size 522284
|
examples/cpp/whisper.wav
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:fdf90cec169e2bf9067baafd918bd24996efb9897484f1dfc1438d916e77f804
|
| 3 |
+
size 411884
|
examples/cpp/zero_shot.wav
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:1c357a82f2f770492f9acc430559a886f585b959d119f9b8279f839898c5b7c0
|
| 3 |
+
size 288044
|
examples/python/add_cough.wav
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:a245f085ad8da2982009b9ca617e08539b72de2785ed0138b3a051d0d6d62c55
|
| 3 |
+
size 501198
|
examples/python/deep_voice.wav
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:b2d6f4ec9a9f8922968636df8d8371e840afbdc0f2a9139b08e997107bc3bb41
|
| 3 |
+
size 316878
|
examples/python/enhance.wav
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:acac796f51c435d6f45e4bc05ec552c583425994fbd20b315922d2ae2ea0d4a1
|
| 3 |
+
size 239118
|
examples/python/get_what_speaker.wav
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:3fc31a22582c4f9f0a0573959f0dd3526052de6df60875ecc6e962acd04777e3
|
| 3 |
+
size 1344078
|
examples/python/happy.wav
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:a5de91d4090d5998addb0fee97d8905561acd6ef13035b189e859e071aa9422c
|
| 3 |
+
size 303438
|
examples/python/increase_volume.wav
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:3feb4e0134dce0356fcd547d1e3f1f68cb8c2486879ab69076a69999a31cd00b
|
| 3 |
+
size 264078
|
examples/python/instruct.wav
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:b2bf530fbdc0ed78f76c7c0d457e28090ef193dfbad47e372fc5fa7da3628255
|
| 3 |
+
size 144078
|
examples/python/raise_pitch.wav
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:b43bb7cc05cd69056c44689a5c5bf04afb23c4736f4edfccc03e2fdf3aa2c547
|
| 3 |
+
size 264078
|
examples/python/replace_lyrics.wav
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:3029ad8e4cf268fd0cadf6d4e930cf9b2417b6ff8de3ab531b6bba958c9b60e9
|
| 3 |
+
size 275598
|
examples/python/replace_words.wav
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:c9d7ad26e4d8879864b7e6fac386d1975f40d80501430a7476ed7fa898135a37
|
| 3 |
+
size 336078
|
examples/python/second_speaker.wav
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:68a50244bdbe4936bcdb20fa14fc12c1ec131ff723d79ed7c1a46205719ba3d7
|
| 3 |
+
size 922638
|
examples/python/speed_1_5x.wav
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:8caa3db0dcd1936709fd527c39e645b539a4c0b65d781d679257d9c3face5166
|
| 3 |
+
size 329358
|
examples/python/standard_mandarin.wav
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:2f9f39ca2a06a5c6b2e8cf0e074e3f68fd8c37a067006d2bbe862f5bfca72589
|
| 3 |
+
size 393678
|
examples/python/vocals_only.wav
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:283d10c3b126353b1dac81bffd460c9c99e7106627dc5ade84748c9b369c4a3f
|
| 3 |
+
size 522318
|
examples/python/whisper.wav
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:55c8f1ee9bd3bd5cf797a065fec5158342c41b929dc13ec9ce8d4251994878d5
|
| 3 |
+
size 411918
|
examples/python/zero_shot.wav
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:30a403cac5b0cba2ae27d80c89a9a439554df701d8ef65046e26eb27705d2e6b
|
| 3 |
+
size 288078
|
qwen2.5-omni-3b-bf16.gguf
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:d1807b5849ac7b814c78b68ad4571aa181aa59280ef69dee222e5f6bb834368d
|
| 3 |
+
size 7447373760
|
qwen2.5-omni-3b-q8_0.gguf
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:732861a7d938d08770f98d4732ee32982ee042c6c7161eed48db5569a8a620c3
|
| 3 |
+
size 4253964480
|
tokenizer/tokenizer.json
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:8441917e39ae0244e06d704b95b3124795cec478e297f9afac39ba670d7e9d99
|
| 3 |
+
size 11421870
|
tokenizer/tokenizer_config.json
ADDED
|
@@ -0,0 +1,223 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"add_prefix_space": false,
|
| 3 |
+
"added_tokens_decoder": {
|
| 4 |
+
"151643": {
|
| 5 |
+
"content": "<|endoftext|>",
|
| 6 |
+
"lstrip": false,
|
| 7 |
+
"normalized": false,
|
| 8 |
+
"rstrip": false,
|
| 9 |
+
"single_word": false,
|
| 10 |
+
"special": true
|
| 11 |
+
},
|
| 12 |
+
"151644": {
|
| 13 |
+
"content": "<|im_start|>",
|
| 14 |
+
"lstrip": false,
|
| 15 |
+
"normalized": false,
|
| 16 |
+
"rstrip": false,
|
| 17 |
+
"single_word": false,
|
| 18 |
+
"special": true
|
| 19 |
+
},
|
| 20 |
+
"151645": {
|
| 21 |
+
"content": "<|im_end|>",
|
| 22 |
+
"lstrip": false,
|
| 23 |
+
"normalized": false,
|
| 24 |
+
"rstrip": false,
|
| 25 |
+
"single_word": false,
|
| 26 |
+
"special": true
|
| 27 |
+
},
|
| 28 |
+
"151646": {
|
| 29 |
+
"content": "<|AUDIO|>",
|
| 30 |
+
"lstrip": false,
|
| 31 |
+
"normalized": false,
|
| 32 |
+
"rstrip": false,
|
| 33 |
+
"single_word": false,
|
| 34 |
+
"special": true
|
| 35 |
+
},
|
| 36 |
+
"151647": {
|
| 37 |
+
"content": "<|audio_bos|>",
|
| 38 |
+
"lstrip": false,
|
| 39 |
+
"normalized": false,
|
| 40 |
+
"rstrip": false,
|
| 41 |
+
"single_word": false,
|
| 42 |
+
"special": true
|
| 43 |
+
},
|
| 44 |
+
"151648": {
|
| 45 |
+
"content": "<|audio_eos|>",
|
| 46 |
+
"lstrip": false,
|
| 47 |
+
"normalized": false,
|
| 48 |
+
"rstrip": false,
|
| 49 |
+
"single_word": false,
|
| 50 |
+
"special": true
|
| 51 |
+
},
|
| 52 |
+
"151649": {
|
| 53 |
+
"content": "<|box_end|>",
|
| 54 |
+
"lstrip": false,
|
| 55 |
+
"normalized": false,
|
| 56 |
+
"rstrip": false,
|
| 57 |
+
"single_word": false,
|
| 58 |
+
"special": true
|
| 59 |
+
},
|
| 60 |
+
"151650": {
|
| 61 |
+
"content": "<|quad_start|>",
|
| 62 |
+
"lstrip": false,
|
| 63 |
+
"normalized": false,
|
| 64 |
+
"rstrip": false,
|
| 65 |
+
"single_word": false,
|
| 66 |
+
"special": true
|
| 67 |
+
},
|
| 68 |
+
"151651": {
|
| 69 |
+
"content": "<|quad_end|>",
|
| 70 |
+
"lstrip": false,
|
| 71 |
+
"normalized": false,
|
| 72 |
+
"rstrip": false,
|
| 73 |
+
"single_word": false,
|
| 74 |
+
"special": true
|
| 75 |
+
},
|
| 76 |
+
"151652": {
|
| 77 |
+
"content": "<|vision_bos|>",
|
| 78 |
+
"lstrip": false,
|
| 79 |
+
"normalized": false,
|
| 80 |
+
"rstrip": false,
|
| 81 |
+
"single_word": false,
|
| 82 |
+
"special": true
|
| 83 |
+
},
|
| 84 |
+
"151653": {
|
| 85 |
+
"content": "<|vision_eos|>",
|
| 86 |
+
"lstrip": false,
|
| 87 |
+
"normalized": false,
|
| 88 |
+
"rstrip": false,
|
| 89 |
+
"single_word": false,
|
| 90 |
+
"special": true
|
| 91 |
+
},
|
| 92 |
+
"151654": {
|
| 93 |
+
"content": "<|vision_pad|>",
|
| 94 |
+
"lstrip": false,
|
| 95 |
+
"normalized": false,
|
| 96 |
+
"rstrip": false,
|
| 97 |
+
"single_word": false,
|
| 98 |
+
"special": true
|
| 99 |
+
},
|
| 100 |
+
"151655": {
|
| 101 |
+
"content": "<|IMAGE|>",
|
| 102 |
+
"lstrip": false,
|
| 103 |
+
"normalized": false,
|
| 104 |
+
"rstrip": false,
|
| 105 |
+
"single_word": false,
|
| 106 |
+
"special": true
|
| 107 |
+
},
|
| 108 |
+
"151656": {
|
| 109 |
+
"content": "<|VIDEO|>",
|
| 110 |
+
"lstrip": false,
|
| 111 |
+
"normalized": false,
|
| 112 |
+
"rstrip": false,
|
| 113 |
+
"single_word": false,
|
| 114 |
+
"special": true
|
| 115 |
+
},
|
| 116 |
+
"151657": {
|
| 117 |
+
"content": "<tool_call>",
|
| 118 |
+
"lstrip": false,
|
| 119 |
+
"normalized": false,
|
| 120 |
+
"rstrip": false,
|
| 121 |
+
"single_word": false,
|
| 122 |
+
"special": false
|
| 123 |
+
},
|
| 124 |
+
"151658": {
|
| 125 |
+
"content": "</tool_call>",
|
| 126 |
+
"lstrip": false,
|
| 127 |
+
"normalized": false,
|
| 128 |
+
"rstrip": false,
|
| 129 |
+
"single_word": false,
|
| 130 |
+
"special": false
|
| 131 |
+
},
|
| 132 |
+
"151659": {
|
| 133 |
+
"content": "<|fim_prefix|>",
|
| 134 |
+
"lstrip": false,
|
| 135 |
+
"normalized": false,
|
| 136 |
+
"rstrip": false,
|
| 137 |
+
"single_word": false,
|
| 138 |
+
"special": false
|
| 139 |
+
},
|
| 140 |
+
"151660": {
|
| 141 |
+
"content": "<|fim_middle|>",
|
| 142 |
+
"lstrip": false,
|
| 143 |
+
"normalized": false,
|
| 144 |
+
"rstrip": false,
|
| 145 |
+
"single_word": false,
|
| 146 |
+
"special": false
|
| 147 |
+
},
|
| 148 |
+
"151661": {
|
| 149 |
+
"content": "<|fim_suffix|>",
|
| 150 |
+
"lstrip": false,
|
| 151 |
+
"normalized": false,
|
| 152 |
+
"rstrip": false,
|
| 153 |
+
"single_word": false,
|
| 154 |
+
"special": false
|
| 155 |
+
},
|
| 156 |
+
"151662": {
|
| 157 |
+
"content": "<|fim_pad|>",
|
| 158 |
+
"lstrip": false,
|
| 159 |
+
"normalized": false,
|
| 160 |
+
"rstrip": false,
|
| 161 |
+
"single_word": false,
|
| 162 |
+
"special": false
|
| 163 |
+
},
|
| 164 |
+
"151663": {
|
| 165 |
+
"content": "<|repo_name|>",
|
| 166 |
+
"lstrip": false,
|
| 167 |
+
"normalized": false,
|
| 168 |
+
"rstrip": false,
|
| 169 |
+
"single_word": false,
|
| 170 |
+
"special": false
|
| 171 |
+
},
|
| 172 |
+
"151664": {
|
| 173 |
+
"content": "<|file_sep|>",
|
| 174 |
+
"lstrip": false,
|
| 175 |
+
"normalized": false,
|
| 176 |
+
"rstrip": false,
|
| 177 |
+
"single_word": false,
|
| 178 |
+
"special": false
|
| 179 |
+
}
|
| 180 |
+
},
|
| 181 |
+
"additional_special_tokens": [
|
| 182 |
+
"<|im_start|>",
|
| 183 |
+
"<|im_end|>",
|
| 184 |
+
"<|AUDIO|>",
|
| 185 |
+
"<|audio_bos|>",
|
| 186 |
+
"<|audio_eos|>",
|
| 187 |
+
"<|box_end|>",
|
| 188 |
+
"<|quad_start|>",
|
| 189 |
+
"<|quad_end|>",
|
| 190 |
+
"<|vision_bos|>",
|
| 191 |
+
"<|vision_eos|>",
|
| 192 |
+
"<|vision_pad|>",
|
| 193 |
+
"<|IMAGE|>",
|
| 194 |
+
"<|VIDEO|>"
|
| 195 |
+
],
|
| 196 |
+
"bos_token": null,
|
| 197 |
+
"chat_template": "{% set audio_count = namespace(value=0) %}{% set image_count = namespace(value=0) %}{% set video_count = namespace(value=0) %}{% for message in messages %}{% if loop.first and message['role'] != 'system' %}<|im_start|>system\nYou are a helpful assistant.<|im_end|>\n{% endif %}<|im_start|>{{ message['role'] }}\n{% if message['content'] is string %}{{ message['content'] }}<|im_end|>\n{% else %}{% for content in message['content'] %}{% if content['type'] == 'image' or 'image' in content or 'image_url' in content %}{% set image_count.value = image_count.value + 1 %}{% if add_vision_id %}Picture {{ image_count.value }}: {% endif %}<|vision_bos|><|IMAGE|><|vision_eos|>{% elif content['type'] == 'audio' or 'audio' in content or 'audio_url' in content %}{% set audio_count.value = audio_count.value + 1 %}{% if add_audio_id %}Audio {{ audio_count.value }}: {% endif %}<|audio_bos|><|AUDIO|><|audio_eos|>{% elif content['type'] == 'video' or 'video' in content %}{% set video_count.value = video_count.value + 1 %}{% if add_vision_id %}Video {{ video_count.value }}: {% endif %}<|vision_bos|><|VIDEO|><|vision_eos|>{% elif 'text' in content %}{{ content['text'] }}{% endif %}{% endfor %}<|im_end|>\n{% endif %}{% endfor %}{% if add_generation_prompt %}<|im_start|>assistant\n{% endif %}",
|
| 198 |
+
"clean_up_tokenization_spaces": false,
|
| 199 |
+
"eos_token": "<|im_end|>",
|
| 200 |
+
"errors": "replace",
|
| 201 |
+
"extra_special_tokens": {
|
| 202 |
+
"image_token": "<|IMAGE|>",
|
| 203 |
+
"audio_token": "<|AUDIO|>",
|
| 204 |
+
"video_token": "<|VIDEO|>",
|
| 205 |
+
"vision_bos_token": "<|vision_bos|>",
|
| 206 |
+
"vision_eos_token": "<|vision_eos|>",
|
| 207 |
+
"audio_bos_token": "<|audio_bos|>",
|
| 208 |
+
"audio_eos_token": "<|audio_eos|>"
|
| 209 |
+
},
|
| 210 |
+
"model_max_length": 32768,
|
| 211 |
+
"pad_token": "<|endoftext|>",
|
| 212 |
+
"processor_class": "Qwen2_5OmniProcessor",
|
| 213 |
+
"split_special_tokens": false,
|
| 214 |
+
"tokenizer_class": "Qwen2Tokenizer",
|
| 215 |
+
"unk_token": null,
|
| 216 |
+
"image_token": "<|IMAGE|>",
|
| 217 |
+
"audio_token": "<|AUDIO|>",
|
| 218 |
+
"video_token": "<|VIDEO|>",
|
| 219 |
+
"vision_bos_token": "<|vision_bos|>",
|
| 220 |
+
"vision_eos_token": "<|vision_eos|>",
|
| 221 |
+
"audio_bos_token": "<|audio_bos|>",
|
| 222 |
+
"audio_eos_token": "<|audio_eos|>"
|
| 223 |
+
}
|