快速上手推理流程
- 通过 pip install airllm 安装包
- 使用 AutoModel.from_pretrained 传入 Hugging Face 模型 ID 或本地路径初始化模型
- 用 model.tokenizer 处理输入文本
- 调用 model.generate 生成输出并解码
开源大模型推理库,用分层流式加载让超大模型跑进消费级显卡
开源大模型推理库,用分层流式加载让超大模型跑进消费级显卡
AirLLM 的定位与常见推理加速框架不同:它不追求高吞吐,而是以牺牲速度换取极低的显存门槛,让没有高端显卡的用户也能运行超大参数模型。选型时需注意:推理前原始模型会被逐层分解并保存到 Hugging Face 缓存目录,需要预留充足磁盘空间;MacOS 上仅支持 Apple 芯片;项目提供 prefetching 等可配置选项。项目以 Apache-2.0 许可开源,通过 pip 安装,处于活跃维护状态。
通过每次仅在 GPU 上保留单层模型的流式加载方式,让 70B 模型在单张 4GB 显存显卡上运行,无需量化、蒸馏或剪枝。
使用 AutoModel.from_pretrained 传入 Hugging Face 模型 ID 或本地路径即可加载,官方称几乎支持所有主流开源大模型。
支持基于分块量化的模型压缩,官方称可将推理速度提升最多约 3 倍,精度损失几乎可忽略。
可在 Apple 芯片的 MacOS 上运行 70B 大模型,需安装 mlx 和 torch。
自 v2.10.1 起支持 CPU 推理以及非分片模型。
在消费级硬件上部署大模型的开发者
在单张 4GB 显存显卡上运行 70B 模型,或在 8GB/12GB 显存上运行 405B/671B 级别模型。
可用状态: 当前可用
价格信息: 开源
支持平台: macOS
价格说明: Apache-2.0 许可证开源项目,可免费通过 pip 安装使用。
最可能的原因是磁盘空间不足。模型切分过程非常消耗磁盘,可扩展磁盘空间、清理 Hugging Face 缓存后重试。
部分模型为受限模型,需要 Hugging Face API token,可在初始化时通过 hf_token 参数提供。
资料核验于 2026-08-31