Hugging Face Blog·· 2023-12-20精选AI 评分75
使用推测解码将 Whisper 推理速度提升 2 倍
Speculative Decoding for 2x Faster Whisper Inference
AI 导读
Hugging Face 博客发文演示了如何在 Whisper 模型中应用推测解码(Speculative Decoding)技术。通过引入 Distil-Whisper 作为辅助模型,在保持转录精度完全一致的前提下,将 16GB T4 GPU 上的推理时间缩短约 2 倍。文章提供了基于 Transformers 库的完整代码实现,并探讨了多语言场景下的模型选择及批处理策略。
推荐理由
文章提供了基于 Transformers 库的完整代码,演示了如何利用 Speculative Decoding 在不损失精度的前提下将 Whisper 推理速度提升约 2 倍。
来源:Hugging Face Blog · huggingface.co