跳到正文
原文
AGI Hunt· ariG23498·· 4 小时前AI 评分47

Hugging Face 集成 TorchTPU:Transformers/Diffusers 原生跑 TPU

Hugging Face 集成 TorchTPU:Transformers/Diffusers 原生跑 TPU

AI 导读

Hugging Face 宣布与 TorchTPU 集成,TPU 由此成为 PyTorch 的“又一种设备”,从 Hub 选模型、to("tpu") 即可运行。Transformers、Diffusers 和 Kernels 已原生支持 TPU,此前跑 PyTorch 需迁移到 torchxla 或学习 JAX,Hugging Face 原有的 Optimum TPU 适配层不再必需。

来源:AGI Hunt · agihunt.info