教程 / 实战精选
Hugging Face 讲解 Accelerate 如何借助 PyTorch 加载和运行超大模型
原始标题:How 🤗 Accelerate runs very large models thanks to PyTorch
内容摘要
Hugging Face 发布博客讲解 Accelerate 如何加载和运行超出单机内存的超大模型。核心做法是用 PyTorch meta device 创建空模型、用 infer_auto_device_map 自动计算 GPU、CPU 与磁盘的设备分配,并按分片加载权重,通过 hooks 在每次前向传播前后搬运权重。
内容分类AI 教程与实战
内容层级精选情报
发布时间(北京时间)
本站收录时间(北京时间)
信息来源Hugging Face:Blog(RSS)
站内情报编号intel-010dae83bfc28473cd77e6d3