跳到正文
原文
Hugging Face Blog·· 2022-09-27精选AI 评分62

Accelerate 如何借助 PyTorch 运行超大模型

How 🤗 Accelerate runs very large models thanks to PyTorch

AI 导读

Hugging Face 介绍 Accelerate 如何借助 PyTorch 的 meta device、自动 device map 和分片 checkpoint,在显存与内存不足的消费级硬件上加载并运行超大语言模型。

推荐理由

原文拆解了 Accelerate 借助 PyTorch meta device 与分片加载运行超大模型的完整流程,可迁移到本地部署实践。

来源:Hugging Face Blog · huggingface.co