Views
No views yet
1pip install --upgrade pip
2pip install --upgrade git+https://github.com/SYSTRAN/faster-whisper datasets[audio]1import torch
2from faster_whisper import WhisperModel
3from datasets import load_dataset
4
5# define our torch configuration
6device = "cuda:0" if torch.cuda.is_available() else "cpu"
7compute_type = "float16" if torch.cuda.is_available() else "float32"
8
9# load model on GPU if available, else cpu
10model = WhisperModel("distil-large-v3", device=device, compute_type=compute_type)
11
12# load toy dataset for example
13dataset = load_dataset("hf-internal-testing/librispeech_asr_dummy", "clean", split="validation")
14sample = dataset[1]["audio"]["path"]
15
16segments, info = model.transcribe(sample, beam_size=1)
17
18for segment in segments:
19 print("[%.2fs -> %.2fs] %s" % (segment.start, segment.end, segment.text))audio argument to transcribe:segments, info = model.transcribe("audio.mp3", beam_size=1)@misc{gandhi2023distilwhisper,
title={Distil-Whisper: Robust Knowledge Distillation via Large-Scale Pseudo Labelling},
author={Sanchit Gandhi and Patrick von Platen and Alexander M. Rush},
year={2023},
eprint={2311.00430},
archivePrefix={arXiv},
primaryClass={cs.CL}
}