Collection of quantized whisper models created by OpenAI • 19 items • Updated • 4
whisper-tiny-FP8-Dynamic
Model Overview
- Model Architecture: whisper-tiny
- Input: Audio-Text
- Output: Text
- Model Optimizations:
- Weight quantization: FP8
- Activation quantization: FP8
- Release Date: 04/16/2025
- Version: 1.0
- Model Developers: Neural Magic
Quantized version of openai/whisper-tiny.
Model Optimizations
This model was obtained by quantizing the weights of openai/whisper-tiny to FP8 data type, ready for inference with vLLM >= 0.5.2.
Deployment
Use with vLLM
This model can be deployed efficiently using the vLLM backend, as shown in the example below.
from vllm.assets.audio import AudioAsset
from vllm import LLM, SamplingParams
# prepare model
llm = LLM(
model="neuralmagic/whisper-tiny-FP8-Dynamic",
max_model_len=448,
max_num_seqs=400,
limit_mm_per_prompt={"audio": 1},
)
# prepare inputs
inputs = { # Test explicit encoder/decoder prompt
"encoder_prompt": {
"prompt": "",
"multi_modal_data": {
"audio": AudioAsset("winning_call").audio_and_sample_rate,
},
},
"decoder_prompt": "<|startoftranscript|>",
}
# generate response
print("========== SAMPLE GENERATION ==============")
outputs = llm.generate(inputs, SamplingParams(temperature=0.0, max_tokens=64))
print(f"PROMPT : {outputs[0].prompt}")
print(f"RESPONSE: {outputs[0].outputs[0].text}")
print("==========================================")
vLLM also supports OpenAI-compatible serving. See the documentation for more details.
Creation
This model was created with llm-compressor by running the code snippet below.
Evaluation
The model was evaluated on LibriSpeech and Fleurs datasets using lmms-eval, via the following commands:
| Benchmark | Split | BF16 | w8a8 | Recovery (%) |
|---|---|---|---|---|
| LibriSpeech (WER) | test-clean | 7.6602 | 7.8941 | 96.53% |
| test-other | 17.1041 | 17.1325 | 98.74% | |
| Fleurs (X→en, WER) | cmn_hans_cn | 43.8226 | 45.0539 | 97.27% |
| en | 13.6638 | 15.2980 | 89.32% | |
| yue_hant_hk | 60.1848 | 67.5437 | 89.10% |
- Downloads last month
- 8
Safetensors
Model size
57.8M params
Tensor type
F32
·
F8_E4M3 ·
Model tree for RedHatAI/whisper-tiny-FP8-Dynamic
Base model
openai/whisper-tiny