跳到正文
Hugging Face Blog·· 14 天前精选AI 评分65

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

Hugging Face ❤️ TorchTPU: TPUs, Natively

AI 导读

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

推荐理由

Hugging Face 官方给出 TorchTPU 的接入方式与实测数据,读者可据此判断 TPU 是否值得纳入现有 PyTorch 工作流。

来源:Hugging Face Blog · huggingface.co