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