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