whisper-medium-FP8-Dynamic

Model Overview

• Model Architecture: whisper-medium
  • 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-medium.

Model Optimizations

This model was obtained by quantizing the weights of openai/whisper-medium 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-medium-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	2.8269	2.8155	100.40%
	test-other	6.4445	6.4124	100.50%
Fleurs (X→en, WER)	cmn_hans_cn	13.3371	13.0153	102.47%
	en	4.6004	4.5386	101.36%
	yue_hant_hk	9.8107	10.2004	96.18%

Base model

openai/whisper-medium

(208)

Collection of quantized whisper models created by OpenAI • 19 items • Updated Apr 30 • 4