跳到正文
Hugging Face Blog·· 2022-09-27AI 评分51

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

How 🤗 Accelerate runs very large models thanks to PyTorch

AI 导读

Hugging Face 官方博客讲解 Accelerate 如何借助 PyTorch 让超大模型在免费 Colab 等消费级硬件上完成加载与推理。文章先指出传统加载流程的内存瓶颈:OPT-6.7B 仅创建模型就需约 26.8GB RAM,BLOOM 或 OPT-176B 这类 176B 参数模型按传统方式需 1.4TB CPU RAM。

来源:Hugging Face Blog · huggingface.co