Whisper-si-experiments
Collection
6 items • Updated
How to use SPEAK-ASR/whisper-si-exp-1 with PEFT:
from peft import PeftModel
from transformers import AutoModelForSeq2SeqLM
base_model = AutoModelForSeq2SeqLM.from_pretrained("openai/whisper-small")
model = PeftModel.from_pretrained(base_model, "SPEAK-ASR/whisper-si-exp-1")How to use SPEAK-ASR/whisper-si-exp-1 with Transformers:
# Load model directly
from transformers import AutoModel
model = AutoModel.from_pretrained("SPEAK-ASR/whisper-si-exp-1", dtype="auto")This model is a fine-tuned version of openai/whisper-small on the Whisper Small - Sinhala ASR Fine-Tuned dataset. It achieves the following results on the evaluation set:
SPEAK-ASR/speak-whisper-small-siMore information needed
More information needed
The following hyperparameters were used during training:
| Training Loss | Epoch | Step | Validation Loss |
|---|---|---|---|
| 0.4531 | 0.6527 | 500 | 0.4665 |
| 0.2969 | 1.3055 | 1000 | 0.2749 |
| 0.2630 | 1.9582 | 1500 | 0.2332 |
| 0.2262 | 2.6110 | 2000 | 0.2186 |
Base model
openai/whisper-small