跳到正文
Hugging Face 官方团队博客· Alvaro Moran·· 18 小时前AI 评分66

Hugging Face 生态通过 TorchTPU 原生支持 TPU

Hugging Face ❤️ TorchTPU: TPUs, Natively

AI 导读

Hugging Face 宣布通过 PyTorch 原生后端 TorchTPU 支持 TPU,Transformers、Diffusers 和 Kernels 可原生运行,模型加载只需 device_map="tpu",不再需要 torch_xla 或 Optimum TPU。

来源:Hugging Face 官方团队博客 · huggingface.co