BuzzASR / swahili

BuzzASR — Swahili

A monolingual automatic speech recognition model for Swahili , fine-tuned from openai/whisper-large-v3 . Part of BuzzASR , a suite of 102 language-specialized ASR models (Findings of EMNLP 2026).

This model uses full fine-tuning (native per-language tokenizer replacement + text multitask fine-tuning) .

Results (normalized CER / WER, %)

~1.8x CER reduction over Whisper zero-shot on the combined test set.

Usage

import torch, torchaudio
from transformers import WhisperForConditionalGeneration, WhisperProcessor

model = WhisperForConditionalGeneration.from_pretrained("BuzzASR/swahili", torch_dtype=torch.float16).to("cuda").eval()
proc  = WhisperProcessor.from_pretrained("BuzzASR/swahili")

wav, sr = torchaudio.load("audio.wav")           # 16 kHz mono
feats = proc(wav[0], sampling_rate=16000, return_tensors="pt").input_features.to("cuda").half()
ids = model.generate(feats, num_beams=1, no_repeat_ngram_size=3, repetition_penalty=1.2)
print(proc.batch_decode(ids, skip_special_tokens=True)[0])

The language/task prompt is baked into the generation config, so no language= argument is needed.

Training data

FLEURS + Common Voice Corpus 25.0 (Mozilla, March 2025; https://commonvoice.mozilla.org/en/datasets ), capped per the paper. Text-only data from the Goldfish corpus (Chang et al., 2026).

Limitations

Monolingual (Swahili only). Evaluated on FLEURS / Common Voice test splits; other domains or dialects may differ.

Citation

Project page: https://lemn-lab.github.io/buzzasr-docs/