OpenASR commited on
Commit
242cab1
Β·
verified Β·
0 Parent(s):

publish redimnet2-b6-cn OpenASR packs

Browse files
.gitattributes ADDED
@@ -0,0 +1 @@
 
 
1
+ *.oasr filter=lfs diff=lfs merge=lfs -text
README.md ADDED
@@ -0,0 +1,105 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: mit
3
+ base_model: PalabraAI/redimnet2
4
+ pipeline_tag: feature-extraction
5
+ library_name: openasr
6
+ tags:
7
+ - speaker-diarization
8
+ - openasr
9
+ - oasr
10
+ ---
11
+
12
+ <div align="center">
13
+
14
+ # ReDimNet2-B6 Speaker Embedder (CN-enhanced) Β· OpenASR
15
+
16
+ **ReDimNet2-B6 speaker embedder for OpenASR diarization β€” 192-d CN-enhanced embeddings, fully on-device**
17
+
18
+ [![License](https://img.shields.io/badge/license-MIT-2563eb.svg)](https://github.com/PalabraAI/redimnet2/blob/main/LICENSE)
19
+ [![Format](https://img.shields.io/badge/format-.oasr-7c3aed.svg)](https://github.com/QuintinShaw/openasr)
20
+ [![Runtime](https://img.shields.io/badge/runtime-OpenASR-111827.svg)](https://openasr.org)
21
+ [![Base model](https://img.shields.io/badge/base-redimnet2-f59e0b.svg)](https://huggingface.co/PalabraAI/redimnet2)
22
+
23
+ Speaker-diarization support pack for the **[OpenASR](https://github.com/QuintinShaw/openasr)** runtime β€”
24
+ pure-Rust inference, **no Python at inference time**.
25
+
26
+ </div>
27
+
28
+ ---
29
+
30
+ ## ✨ Highlights
31
+
32
+ - πŸ—£οΈ **Preferred OpenASR speaker embedder** β€” when installed, diarization prefers this pack over WeSpeaker for anonymous speaker labels on any ASR family
33
+ - 🧬 **192-dim ReDimNet2-B6** β€” PalabraAI's dimension-reshaping speaker net (12.5M params) with a Chinese-enhanced vb2+vox2+cnc2 training mix
34
+ - πŸ”’ **Diarization, not identification** β€” anonymous session-relative labels; embeddings stay local and are discarded after the request unless you explicitly enroll a local profile
35
+ - 🎯 **Parity-gated packaging** β€” ggml-graph forward pass matches the upstream Python reference at cosine β‰₯ 0.9999 on held-out fixtures
36
+ - πŸ¦€ **Native in OpenASR** β€” `.oasr` packs run with no Python at inference, engineered for peak performance on CPU & GPU
37
+
38
+ ## πŸš€ Quickstart
39
+
40
+ ```bash
41
+ # 1. Install the OpenASR CLI Β· https://openasr.org
42
+ # 2. Pull the pack
43
+ openasr pull redimnet2-b6-cn:fp16
44
+
45
+ # 3. Diarize any transcription (works with every OpenASR ASR model)
46
+ openasr transcribe meeting.wav --model xasr-zh-en --diarize --format srt
47
+ ```
48
+
49
+ ## πŸ“¦ Pack
50
+
51
+ | Quant | File (`.oasr`) | Size |
52
+ |:------|:---------------|-----:|
53
+ | fp16 | `redimnet2-b6-cn-fp16.oasr` | 28 MB |
54
+ | q8_0 | `redimnet2-b6-cn-q8_0.oasr` | 27 MB |
55
+ | f32 | `redimnet2-b6-cn-f32.oasr` | 50 MB |
56
+
57
+ <sub>Single raw-**f32** build: the pure-Rust forward pass consumes f32 directly and the
58
+ parity gates assert bit-exact outputs vs the upstream weights, so no integer
59
+ quantization is produced.</sub>
60
+
61
+ ## 🧠 About ReDimNet2-B6 Speaker Embedder (CN-enhanced)
62
+
63
+ ReDimNet2-B6 is PalabraAI's 12.5M-parameter speaker-embedding model from the
64
+ ReDimNet2 family, trained on a VoxBlink2 + VoxCeleb2 + CN-Celeb2 mix so English and
65
+ Chinese speakers share one embedding space. OpenASR packages the MIT-licensed
66
+ checkpoint as a local `.oasr` capability pack and runs it through a ggml graph
67
+ (not a pure-Rust hand-written forward). When this pack is installed it becomes the
68
+ preferred speaker-embedding stage for diarization; the legacy WeSpeaker pack remains
69
+ available as a fallback until a later quality gate retires it. Embeddings are 192-d
70
+ cosine vectors with a ReDimNet-specific calibration profile, separate from WeSpeaker's
71
+ 256-d space.
72
+
73
+ ## βš™οΈ How this pack was made
74
+
75
+ Converted from [PalabraAI/redimnet2](https://huggingface.co/PalabraAI/redimnet2) with the OpenASR importer:
76
+
77
+ ```bash
78
+ openasr model-pack import redimnet2 <src>.safetensors <out>.oasr \
79
+ --package-id redimnet2-b6-cn
80
+ ```
81
+
82
+ The `.oasr` container is GGUF-backed; every tensor is stored as raw f32 so the
83
+ pack round-trips bit-identically against the source weights.
84
+
85
+ ## βš–οΈ License
86
+
87
+ This pack **inherits the upstream model's license: MIT**
88
+ ([source](https://github.com/PalabraAI/redimnet2/blob/main/LICENSE)). OpenASR packaging retains the upstream copyright;
89
+ the only modification is format conversion.
90
+
91
+ ## πŸ™ Acknowledgements
92
+
93
+ This pack redistributes **PalabraAI/redimnet2** checkpoint
94
+ `b6-vb2+vox2+cnc2_v0-lm.pt` in OpenASR's `.oasr` runtime format. Credit for the
95
+ model architecture, training, and original weights belongs to the upstream
96
+ PalabraAI / ReDimNet2 authors (paper: ReDimNet2: Scaling Speaker Verification via
97
+ Time-Pooled Dimension Reshaping). The upstream model is licensed under **MIT**;
98
+ OpenASR packaging retains that license and attribution, with the only modification
99
+ being format conversion for local ggml-graph loading.
100
+
101
+ ## πŸ”— Links
102
+
103
+ - πŸ¦€ **OpenASR** β€” <https://github.com/QuintinShaw/openasr>
104
+ - 🌐 **Website** β€” <https://openasr.org>
105
+ - πŸ€— **Upstream model** β€” [PalabraAI/redimnet2](https://huggingface.co/PalabraAI/redimnet2)
redimnet2-b6-cn-f32.oasr ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:34f38adb26892a04b9bd39a3abc3de0606c3e406c133c25e20579d8547917678
3
+ size 50048672
redimnet2-b6-cn-fp16.oasr ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b889eb50978957fcf5dae9dbcdb9aca7d3834bc72c3680c01e666b2d7e218727
3
+ size 27775264
redimnet2-b6-cn-q8_0.oasr ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d489dede665dfbaaa7a35d38055dc3eb4df7d6bb8fc8cad625d8332d845c1db8
3
+ size 26738464