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