跳到正文
Hugging Face Blog·· 2 天前精选AI 评分76

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

Hugging Face ❤️ TorchTPU: TPUs, Natively

AI 导读

Hugging Face 宣布接入 PyTorch 原生 TPU 后端 TorchTPU,Transformers、Diffusers 和 Kernels 可直接在 TPU 上运行,不再需要 torch_xla 或 Optimum TPU。

推荐理由

原文给出安装命令与各库接入方式,读者可据此判断 PyTorch 模型迁移到 TPU 的改动量。

来源:Hugging Face Blog · huggingface.co