CoolFace
Modelpublic

Sa1Krishna/sema-whisper-small-springlab-hindi-finetuned

sourceHugging Faceapache-2.0updated 6mo agoView on Hugging Face
0likes
Model Card

Whisper Small โ€” SPRINGLab Hindi Fine-tuned ๐ŸŽ™๏ธ

Fine-tuned version of openai/whisper-small for Hindi automatic speech recognition using LoRA (PEFT).

Model Details

ParameterValue
Base Modelopenai/whisper-small
DatasetSPRINGLab/IndicVoices-R_Hindi
Train Samples25,002
Eval Samples1,316
Training Epochs3
Training Steps2,346
Best Checkpointcheckpoint-2346
Best Eval Loss0.2637
Best Eval WER26.52
20-sample Base WER59.44
20-sample FT WER20.85
LoRA Rank16
LoRA Alpha32
LoRA Dropout0.05
LoRA Targetsqproj, vproj
Learning Rate5e-5
Train Batch Size8
Grad Accumulation4
Effective Batch32
Precisionbfloat16
HardwareGoogle Colab A100
MethodLoRA fine-tuning with PEFT

Validation Summary

The final selected checkpoint was checkpoint-2346, which was manually evaluated after training and slightly outperformed checkpoint-2000 on the full validation split.

  • โ€”checkpoint-2000 eval WER: 26.6488
  • โ€”checkpoint-2346 eval WER: 26.52

Usage

python
from transformers import pipeline

asr = pipeline(
    task='automatic-speech-recognition',
    model='Sa1Krishna/sema-whisper-small-springlab-hindi-finetuned',
    device=0
)

result = asr(
    'hindi_audio.wav',
    generate_kwargs={
        'language': 'hindi',
        'task': 'transcribe'
    }
)

print(result['text'])

Training Details

Trained on SPRINGLab/IndicVoices-R_Hindi using a 95/5 train-validation split and Hindi normalized transcripts.

Training Config

  • โ€”Framework: Hugging Face Transformers + PEFT
  • โ€”Fine-tuning method: LoRA
  • โ€”Precision: bfloat16
  • โ€”Learning rate: 5e-5
  • โ€”Batch size: 8
  • โ€”Gradient accumulation: 4
  • โ€”Evaluation cadence: every 500 steps
  • โ€”Sanity check: 20-sample qualitative comparison at step 200

Notes

  • โ€”Fine-tuned for Hindi speech recognition.
  • โ€”Uses the multilingual Whisper tokenizer and decoder for Hindi transcription.
  • โ€”Final checkpoint was chosen using full-validation WER plus qualitative review.

Limitations

  • โ€”Optimized for Hindi ASR only.
  • โ€”May still struggle with heavy accents, rare proper nouns, and unusual numerals.
  • โ€”Performance may vary on domains very different from IndicVoices-R.