whisper-medium.en
ONNX builds of openai/whisper-medium.en.
Whisper is OpenAI's automatic speech recognition model. This checkpoint is English-only: it was trained on the English subset, which makes it more accurate on English than the multilingual checkpoint of the same size, and unusable on anything else.
It reads audio in 30-second windows and decodes text one token at a time, and produces segment-level timestamps.
This checkpoint: 769 M parameters β close to large accuracy at half the size.
Variants
Every variant is the same checkpoint at a different weight precision. Pick a row, download that folder, and read onnx_manifest.json in it for the graph names and their checksums.
| Variant | Size | vs fp32 |
WER | vs fp32 |
|---|---|---|---|---|
fp16 |
2,554 MB | Γ·1.3 | β | β |
plain-int8 |
866 MB | Γ·3.8 | 2.50 % | -0.21 pt |
plain-int4 |
488 MB | Γ·6.7 | 2.19 % | -0.52 pt |
sym-int4-asym |
490 MB | Γ·6.7 | 2.08 % | -0.63 pt |
zone-int4-attn |
630 MB | Γ·5.2 | 1.46 % | -1.25 pt |
fp32 is not published for this checkpoint β it is the source format, available from the original repository. It is used above as the reference the other rows are measured against.
Word error rate measured on ten minutes of English speech against a hand-checked transcript, greedy decoding, no voice-activity detection. It is a single corpus in a single language: use it to compare the rows against each other, not as an absolute score.
What the names mean
fp16β Half-precision encoder, full-precision decoder. Smaller thanfp32and slower on CPU, where no float16 kernel exists β useful on GPU.plain-int8β 8-bit weights, nothing excluded.plain-int4β 4-bit weights, nothing excluded. Usually the smallest that still works.sym-int4-asymβ 4-bit weights with an asymmetric scale, which does not assume the weights are centred on zero.zone-int4-attnβ 4-bit weights, with the attention projections kept at 8 bits.
The int4 variants are meant for the CPU. On ONNX Runtime's CUDA provider the 4-bit matrix kernel (MatMulNBits) loses precision above 128 rows, which covers every encoder pass.
Licence
These are format conversions. The weights, and the licence they come with, are those of the original checkpoint β see the source repository.
Model tree for Younicorn/whisper-medium.en
Base model
openai/whisper-medium.en