Views
No views yet
Quantized with INT8 dynamic quantization (AutoQuantizationConfig.avx2,is_static=False,per_channel=False).
| Property | Value |
|---|---|
| Base architecture | openai/whisper-base |
| Fine-tuned model | Saidakbar01/whisper-base-uz-finetuned |
| Format | ONNX (INT8) |
| Size | 170 MB |
| WER (200-sample validation) | 56.31% |
| Language | Uzbek (uz) |
| Task | Automatic Speech Recognition |
encoder_model_quantized.onnxdecoder_model_quantized.onnxdecoder_with_past_model_quantized.onnx1from optimum.onnxruntime import ORTModelForSpeechSeq2Seq
2from transformers import WhisperProcessor
3import torch
4
5# Load processor and ONNX model
6processor = WhisperProcessor.from_pretrained(
7 "openai/whisper-base",
8 language="uz",
9 task="transcribe",
10)
11model = ORTModelForSpeechSeq2Seq.from_pretrained(
12 "Saidakbar01/whisper-base-uz-onnx-int8",
13 encoder_file_name="encoder_model_quantized.onnx",
14 decoder_file_name="decoder_model_quantized.onnx",
15 decoder_with_past_file_name="decoder_with_past_model_quantized.onnx",
16)
17
18# Transcribe audio (numpy float32 array at 16 kHz)
19inputs = processor(audio_array, sampling_rate=16000, return_tensors="pt")
20with torch.no_grad():
21 predicted_ids = model.generate(
22 inputs.input_features,
23 language="uz",
24 task="transcribe",
25 )
26transcription = processor.batch_decode(predicted_ids, skip_special_tokens=True)[0]
27print(transcription)DavronSherbaev/uzbekvoice-filtered dataset.
WER was measured on 200 samples from the validation split using jiwer.| Model | WER |
|---|---|
| openai/whisper-base (baseline) | 200.69% |
| Saidakbar01/whisper-base-uz-finetuned (fine-tuned) | 53.13% |
| whisper-base-uz-onnx-int8 (this model) | 56.31% |