跳到正文
Hugging Face 社区博客·· 2022-09-27AI 评分71

Hugging Face 讲解 Accelerate 如何借助 PyTorch 加载和运行超大模型

How 🤗 Accelerate runs very large models thanks to PyTorch

AI 导读

Hugging Face 发布博客讲解 Accelerate 如何加载和运行超出单机内存的超大模型。核心做法是用 PyTorch meta device 创建空模型、用 infer_auto_device_map 自动计算 GPU、CPU 与磁盘的设备分配,并按分片加载权重,通过 hooks 在每次前向传播前后搬运权重。

来源:Hugging Face 社区博客 · huggingface.co