(lo)whipa-models
Collection
Full and PEFT LoRA (LoWhIPA) fine-tuned Whisper-base and Whisper-large-v2 models for language-agnostic IPA transcription of speech.
•
14 items
•
Updated
This Whisper-for-IPA (WhIPA) model adapter is a PEFT LoRA fine-tuned version of openai/whisper-base on a subset of:
For deployment and description, please refer to https://github.com/jshrdt/whipa.
from transformers import WhisperForConditionalGeneration, WhisperTokenizer, WhisperProcessor
from peft import PeftModel
tokenizer = WhisperTokenizer.from_pretrained("openai/whisper-base", task="transcribe")
tokenizer.add_special_tokens({"additional_special_tokens": ["<|ip|>"] + tokenizer.all_special_tokens})
base_model = WhisperForConditionalGeneration.from_pretrained("openai/whisper-base")
base_model.generation_config.lang_to_id["<|ip|>"] = tokenizer.convert_tokens_to_ids(["<|ip|>"])[0]
base_model.resize_token_embeddings(len(tokenizer))
whipa_model = PeftModel.from_pretrained(base_model, "jshrdt/lowhipa-base-comb")
whipa_model.generation_config.language = "<|ip|>"
whipa_model.generation_config.task = "transcribe"
whipa_processor = WhisperProcessor.from_pretrained("openai/whisper-base", task="transcribe")
More information needed
More information needed
| Training Loss | Epoch | Validation Loss |
|---|---|---|
| 1.5428 | 2.0323 | 1.2981462478637695 |
| 0.7498 | 4.0645 | 0.8457677960395813 |
| 0.5968 | 6.0968 | 0.759925901889801 |
| 0.5156 | 8.1290 | 0.7213243246078491 |
| 0.4603 | 10.1613 | 0.7064764499664307 |
Base model
openai/whisper-base