跳到正文
原文
Hugging Face Blog·· 2023-05-11精选AI 评分68

Hugging Face 发布 Assisted Generation 解码方法,文本生成延迟最多降低 10 倍

Assisted Generation: a new direction toward low-latency text generation

AI 导读

Hugging Face 在 Transformers 中推出 Assisted Generation 解码方法,用小模型生成候选 token、大模型前向传播验证,降低自回归文本生成延迟。

推荐理由

原文解释了文本生成延迟来自内存带宽而非计算,并给出 Assisted Generation 的原理、代码和延迟数据,方法可直接复现。

来源:Hugging Face Blog · huggingface.co