Hugging Face Blog·· 2023-05-11精选AI 评分78
Hugging Face 推出辅助生成技术降低文本生成延迟
Assisted Generation: a new direction toward low-latency text generation
AI 导读
Hugging Face 在 Transformers 库中引入辅助生成(Assisted Generation)技术,利用小模型生成候选 token 并由大模型验证,从而减少前向传播次数。该方法在模型显存不足需卸载时最高可实现 10 倍加速,在显存充足时也可获得 2 至 3 倍加速,且支持贪婪解码与采样。
推荐理由
Hugging Face 在 Transformers 中引入辅助生成技术,通过小模型辅助大模型解码,显著降低文本生成延迟。
来源:Hugging Face Blog · huggingface.co