Hugging Face Blog·· 2022-09-27AI 评分
Hugging Face Accelerate 如何借助 PyTorch 运行超大模型
Hugging Face 的 Accelerate 利用 PyTorch 的 meta device 先创建无权重空模型,再逐层加载权重并按设备映射分配,从而在内存和单卡显存不足时也能运行超大模型。
当前提供 AI 导读,完整正文请阅读原文。
来源:Hugging Face Blog · huggingface.co
Hugging Face 的 Accelerate 利用 PyTorch 的 meta device 先创建无权重空模型,再逐层加载权重并按设备映射分配,从而在内存和单卡显存不足时也能运行超大模型。
当前提供 AI 导读,完整正文请阅读原文。
来源:Hugging Face Blog · huggingface.co