跳到正文
Hugging Face Blog·· 18 小时前精选AI 评分73

Hugging Face 接入 TorchTPU,Transformers 与 Diffusers 原生支持 TPU

Hugging Face ❤️ TorchTPU: TPUs, Natively

AI 导读

Hugging Face 宣布 TorchTPU 作为 PyTorch 原生后端接入其生态,Transformers、Diffusers 和 Kernels 可直接在 TPU 上运行,加载模型时传 device_map="tpu" 即可,不再需要 Optimum TPU。

推荐理由

Hugging Face 官方说明 TorchTPU 让 Transformers 与 Diffusers 直接以 tpu 为设备运行,并给出实测加速数据。

来源:Hugging Face Blog · huggingface.co