whisper-large-v2-quantized.w8a8

Model Overview

• Model Architecture: whisper-large-v2
  • Input: Audio-Text
  • Output: Text
• Model Optimizations:
  • Weight quantization: INT8
  • Activation quantization: INT8
• Release Date: 04/16/2025
• Version: 1.0
• Model Developers: Neural Magic

Quantized version of openai/whisper-large-v2.

Model Optimizations

This model was obtained by quantizing the weights of openai/whisper-large-v2 to INT8 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-large-v2-quantized.w8a8",
 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:

Accuracy

Benchmark	Split	BF16	w8a8	Recovery (%)
LibriSpeech (WER)	test-clean	3.1437	3.1343	100.30%
	test-other	5.2362	5.2021	100.66%
Fleurs (X→en, WER)	cmn_hans_cn	15.2148	15.4498	98.48%
	en	4.0717	4.0717	100.00%
	yue_hant_hk	8.5106	8.3830	101.52%

Base model

openai/whisper-large-v2

(9)

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