Перейти к содержимому

Whisper Gets 2x Faster by Adding a Second Model, Not a Smaller One

Kemmu Draws Tech

0:00 / 0:00

Whisper Gets 2x Faster by Adding a Second Model, Not a Smaller One

12 просмотров · 11 дней назад
Kemmu Draws Tech
9 подписчиков
12 просмотров · 11 дней назад
Hugging Face pairs Whisper large-v2 with a two-layer Distil-Whisper draft model and gets roughly double the transcription speed with byte-identical output. We treat that as a roofline problem: weigh the decoder, divide by the A100's memory bandwidth, price a draft step, and solve for the acceptance rate the advertised 2x actually requires. Chapters 0:00 Six minutes for an hour 1:20 Bytes moved, or math done 3:35 One operation per byte 4:38 Two layers instead of thirty two 6:50 Solving for the acceptance rate 8:47 Wrong, cheaply Sources Speculative Decoding for 2x Faster Whisper Inference — https://huggingface.co/blog/whisper-s... Fast Inference from Transformers via Speculative Decoding — https://arxiv.org/abs/2211.17192 Distil-Whisper: Robust Knowledge Distillation via Large-Scale Pseudo Labelling — https://arxiv.org/abs/2311.00430 GitHub - openai/whisper: Robust Speech Recognition via Large-Scale Weak Supervision · GitHub — https://github.com/openai/whisper NVIDIA A100 | NVIDIA — https://www.nvidia.com/en-us/data-cen... Drawn with Inkstack.