smdesai commited on
Commit
aa24cce
·
verified ·
1 Parent(s): 3002858

Add model card

Browse files
Files changed (1) hide show
  1. README.md +89 -0
README.md ADDED
@@ -0,0 +1,89 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: cc-by-4.0
3
+ base_model: nvidia/canary-1b-v2
4
+ pipeline_tag: automatic-speech-recognition
5
+ library_name: coreml
6
+ language:
7
+ - bg
8
+ - hr
9
+ - cs
10
+ - da
11
+ - nl
12
+ - en
13
+ - et
14
+ - fi
15
+ - fr
16
+ - de
17
+ - el
18
+ - hu
19
+ - it
20
+ - lv
21
+ - lt
22
+ - mt
23
+ - pl
24
+ - pt
25
+ - ro
26
+ - sk
27
+ - sl
28
+ - es
29
+ - sv
30
+ - ru
31
+ - uk
32
+ tags:
33
+ - coreml
34
+ - apple-neural-engine
35
+ - on-device
36
+ - ios
37
+ - macos
38
+ - speech
39
+ - audio
40
+ - automatic-speech-recognition
41
+ - automatic-speech-translation
42
+ - canary
43
+ - fastconformer
44
+ - nemo
45
+ - kv-cache
46
+ - fp16
47
+ ---
48
+ # Canary 1B v2 — CoreML (FP16, KV-cache)
49
+
50
+ On-device CoreML port of NVIDIA **canary-1b-v2** for Apple silicon (iOS / macOS): FP32 preprocessor, FP16 Apple Neural Engine-resident FastConformer encoder, and a stateful KV-cache Transformer decoder driven by a host-side greedy loop. This is the reference (uncompressed) build; the memory-optimized variants below are derived from it and match its accuracy.
51
+
52
+ **Base model:** [nvidia/canary-1b-v2](https://huggingface.co/nvidia/canary-1b-v2) (NVIDIA NeMo `EncDecMultiTaskModel`, 1B parameters, 25 European languages, ASR + speech translation). **License:** the base model is released under [CC-BY-4.0](https://creativecommons.org/licenses/by/4.0/legalcode.en); this conversion carries the same license. Please attribute NVIDIA for the model.
53
+
54
+ ## This build
55
+
56
+ | | |
57
+ |---|---|
58
+ | Encoder | FP16 weights, Neural Engine-resident, 1.58 GB |
59
+ | Decoder / cross-KV | FP16, 271 MB + 34 MB |
60
+ | Download | 1.89 GB |
61
+ | iOS RAM while transcribing | 1.9 GB (measured, 6-minute file) |
62
+
63
+ ## Build family
64
+
65
+ All 1B v2 builds share the same preprocessor, tokenizer, package layout, and decode contract; only the weight format of the encoder and/or decoder differs. Download sizes are as hosted on the Hub; iOS RAM was measured in an app while transcribing a 6-minute file, or predicted from the Neural Engine-resident weight region where marked.
66
+
67
+ | Repo | Build | Encoder weights | Decoder weights | Download | iOS RAM |
68
+ |---|---|---|---|---|---|
69
+ | **`canary-1b-v2-coreml` (this repo)** | FP16 (reference) | FP16 | FP16 | 1.89 GB | 1.9 GB (measured) |
70
+ | [`canary-1b-v2-int8-coreml`](https://huggingface.co/smdesai/canary-1b-v2-int8-coreml) | INT8 encoder | INT8 per-channel | FP16 | 1.10 GB | 1.2 GB (measured) |
71
+ | [`canary-1b-v2-pal6-coreml`](https://huggingface.co/smdesai/canary-1b-v2-pal6-coreml) | 6-bit encoder | 6-bit palette, g=16 | FP16 | 0.92 GB | 1.0 GB (measured) |
72
+ | [`canary-1b-v2-int8full-coreml`](https://huggingface.co/smdesai/canary-1b-v2-int8full-coreml) | INT8 full | INT8 per-channel | INT8 per-channel | 0.95 GB | ~1.07 GB (predicted) |
73
+ | [`canary-1b-v2-pal6-int8-coreml`](https://huggingface.co/smdesai/canary-1b-v2-pal6-int8-coreml) | 6-bit + INT8 | 6-bit palette, g=16 | INT8 per-channel | 0.77 GB | ~0.88 GB (predicted) |
74
+ | [`canary-180m-flash-coreml`](https://huggingface.co/smdesai/canary-180m-flash-coreml) | 180M Flash | FP16 (17L, d=512) | FP16 (4L) | 0.37 GB | — |
75
+
76
+ ## Files
77
+
78
+ | File | Contents |
79
+ |---|---|
80
+ | `canary_preprocessor.mlmodelc` | FP32 mel front end (128 mel bins, 16 kHz mono, 15 s window) |
81
+ | `canary_encoder.mlmodelc` | FastConformer encoder, 32 layers, d=1024, subsampling 8 |
82
+ | `canary_cross_kv.mlmodelc` | Cross-attention K/V projection, 8 layers (run once per window) |
83
+ | `canary_decoder_kv.mlmodelc` | Stateful single-step Transformer decoder, 8 layers, 16384-way LM head |
84
+ | `canary_spe.model` | SentencePiece tokenizer (16384 pieces) |
85
+ | `metadata.json` | Shapes, decode geometry, seed tokens, and the weight recipe of this build |
86
+
87
+ ## Credits
88
+
89
+ Model: NVIDIA NeMo team, [canary-1b-v2](https://huggingface.co/nvidia/canary-1b-v2), CC-BY-4.0.