Views
No views yet
openai/whisper-small (244M) fine-tuned on
the YECS Yoruba–English code-switching corpus for plain transcription (no language
tags). It is the control for the language-tag-injection A/B study — its tagged twin,
LyngualLabs/yecs-asr-whisper-lid,
adds 99.54% per-word language ID at no WER cost.| Metric | This model | whisper-small-yoruba (reported) |
|---|---|---|
| WER — tone-aware | 16.69% | 20.76% |
| WER — tone-insensitive | 13.83% | — |
| CER — tone-aware | 6.37% | — |
| CER — tone-insensitive | 5.09% | — |
1from transformers import WhisperForConditionalGeneration, WhisperProcessor
2import torch, librosa
3
4repo = "LyngualLabs/yecs-asr-whisper-plain"
5proc = WhisperProcessor.from_pretrained(repo)
6model = WhisperForConditionalGeneration.from_pretrained(repo, torch_dtype=torch.bfloat16).to("cuda").eval()
7
8audio, _ = librosa.load("utterance.wav", sr=16000)
9feats = proc.feature_extractor(audio, sampling_rate=16000, return_tensors="pt").input_features
10ids = model.generate(feats.to("cuda", torch.bfloat16),
11 language="yoruba", task="transcribe", max_new_tokens=225)
12print(proc.tokenizer.batch_decode(ids, skip_special_tokens=True)[0])openai/whisper-small · 5 epochs · lr 1e-5 · bf16 · effective batch 32 · 1×H200.yoruba, task transcribe.afroscale_ai_cmu_africa/yecs-lid, run whisper-plain.yo, en).tag-injection/)openai/whisper-small). YECS corpus terms apply to the
training data.