Whisper medium β€” 30 s TFLite (LiteRT), quantized

Quantized TFLite (LiteRT) exports of openai/whisper-medium: 30 s fixed-window graphs split into encode and decode signatures, matching the graph interface of litert-community/whisper-tiny and litert-community/whisper-base β€” drop-in for pipelines built against those graphs.

Exported by the LiteRT-LM-Unity project from the openai checkpoint (transformers TFWhisperForConditionalGeneration β†’ two-signature 30 s graph), then post-training quantized. f32 source exports (~3 GB) are retained by the producing project and available on request.

Files

File Size Recipe
whisper_medium_30s_i8.tflite 794 MB dynamic-range int8 weights (channelwise), fp32 activations
whisper_medium_30s_i4.tflite 634 MB dynamic_wi4b64_afp32 mixed ("L1"): int4 blockwise-64 weights (fp16 scales) with the token-embedding/logits scopes kept at int8 channelwise

Integration notes

  • 80 mel bins / vocab 51865 β€” whisper-medium uses the classic Whisper frontend and token ids (same as tiny/base), not the 128-mel/51866 large-v3 family layout.
  • Two signatures: encode (log-mel spectrogram, 30 s window β†’ encoder hidden states) and decode (token ids + encoder states β†’ logits). The decoder is a fixed-length full re-run per step (no KV cache), matching the tiny/base graph interface.
  • Tokenizer: use tokenizer.json from openai/whisper-medium (differs from whisper-base's tokenizer).

Quantization

Quantizer: ai-edge-quantizer 0.8.0, post-training dynamic-range.

  • i8: int8 weights channelwise, fp32 activations (dynamic_wi8_afp32 scheme).
  • i4: dynamic_wi4b64_afp32 (int4 weights, blockwise-64, fp16 scales) with int8 overrides on the token-embedding/logits table. Pure full-scope int4 was tested and rejected (Korean transcription errors); an additional encoder-at-int8 variant produced identical transcripts and was discarded for size. int4 channelwise and blockwise-32 recipes are known-bad (quality collapse / immediate EOS) and were not used.

Validation

9-clip Korean/English test set (Korean tactical-report sentence "2025λ…„ 3μ›” 5일 μ „μˆ ν‰κ°€ κ²°κ³Ό 보고", English equivalent, weather/status sentences, short Korean voice commands). Greedy decode, desktop CPU (XNNPACK, 8 threads). CER against punctuation-normalized references (space-removed); "exact" = normalized exact match.

Variant Exact /9 CER ko CER en Avg RTF Avg ms/step
i8 7 0.042 0.000 1.67 295
i4 7 0.042 0.000 1.74 311

i4 reproduces i8's transcripts on every clip. The only content error in either tier is one short voice-command clip heard as 음ν–₯ 증가 instead of μŒλŸ‰ 증가 (CER 0.250 on that clip); all other misses vs exact are spacing-only or filename artifacts.

Runtime compatibility

Validated with the LiteRT (ai-edge-litert) interpreter on Windows x86_64 CPU (XNNPACK) and within the LiteRT-LM-Unity v0.14.0 pipeline (LiteRT-LM v0.14.0, Windows x86_64 + Android arm64, Snapdragon 865-class device).

Caveats

  • 30 s fixed window; decoder is a full-sequence re-run per step (no KV cache) to match the litert-community tiny/base interface β€” a KV-cached runtime will be substantially faster per token.
  • Known content weakness: μŒλŸ‰ β†’ 음ν–₯ confusion on one Korean voice-command clip (both tiers; same confusion as whisper-base i4).
  • Language forcing recommended for short clips (e.g. <|ko|> / <|en|>).

Produced by

LiteRT-LM-Unity (release v0.14.0-unity). Whisper weights are MIT (OpenAI); quantized derivatives inherit the base license.

Downloads last month
75
Inference Providers NEW
This model isn't deployed by any Inference Provider. πŸ™‹ Ask for provider support

Model tree for litert-community/whisper-medium

Finetuned
(903)
this model