跳到正文
原文
Hugging Face Blog·· 2022-07-27AI 评分40

Hugging Face:用 TensorFlow 和 XLA 加速文本生成,最高快 100 倍

Faster Text Generation with TensorFlow and XLA

AI 导读

Hugging Face 的 transformers 库在 TensorFlow 下现可通过 XLA 编译文本生成,速度最高提升 100 倍,甚至快于 PyTorch。该功能需 transformers >= 4.21.0,支持采样、贪心解码、Beam Search 及 temperature、max_new_tokens 等参数控制。

来源:Hugging Face Blog · huggingface.co