通过一份 YAML 配置完成 LLM 微调与后训练的命令行工具,支持在低显存消费级显卡上训练。
lyogavin/airllm
Python 库,逐层加载大语言模型,让 70B 级模型可在单张 4GB 显卡上推理,无需量化。
- Star
- 39,483
- Fork
- 4,026
- 未关闭 Issue
- 167
- 最近提交
- 昨天
这是什么?
AirLLM 是一个 Python 库,用于在显存很小的 GPU 上运行大语言模型。它不会一次性把整个模型装入显存,而是每次只在 GPU 上保留一个解码层,因此所需显存取决于单层大小,而不是模型总参数量。项目称可在约 4GB 显存上运行 Llama 3.x 70B,在约 12GB 上运行 DeepSeek-V3(671B),并支持 Qwen3-235B、Kimi K3 等新模型在单卡上推理。
用户通过统一的 AutoModel.from_pretrained 加载模型,可传入 Hugging Face 仓库 ID 或本地路径。可选的分块 4bit 或 8bit 压缩能减少磁盘加载耗时并提升推理速度。库还支持基于 MLX 的 Apple 芯片 macOS 运行和 CPU 推理。近期版本加入了基于 LoRA 的微调,冻结的基础权重从磁盘按层流式读取。
主要代价是速度:每次前向都要从磁盘读取各层,因此需要先把原始模型切分成逐层文件,并预留较大的磁盘空间。
核心功能
- 01逐层加载,70B 级模型可在单张 4GB 显存的 GPU 上运行。
- 02通过统一的 AutoModel 接口加载 Hugging Face 仓库或本地模型。
- 03可选 4bit 和 8bit 分块压缩,加快加载速度。
- 04支持 Apple 芯片 macOS(基于 MLX)和 CPU 推理。
- 05提供 LoRA 微调,冻结的基础权重从磁盘流式读取。
适用场景
- 在显存有限的消费级显卡上体验 Llama 70B 等大型开源模型。
- 在单台工作站显卡上运行 DeepSeek-V3 这类超大 MoE 模型。
- 在小显存 GPU 上使用 LoRA 微调大模型。
- 在 Apple 芯片 Mac 上本地运行大模型。
快速上手
- 1
安装 airllm 包
pip install airllm - 2
通过 Hugging Face 仓库 ID 加载模型并运行推理(参见 README 中的快速开始代码)
model = AutoModel.from_pretrained("Qwen/Qwen3-32B")
使用前须知
- 推理前会先把原始模型拆分保存为分层分片,因此需要在 Hugging Face 缓存目录中有足够的磁盘空间。
- 部分较新的模型需要特定版本的 transformers 或额外依赖,例如 Kimi K3 需要 compressed-tensors 和 flash-attn。
- macOS 仅支持 Apple 芯片,并且需要安装 mlx 和 torch。
标签
常见问题
- airllm 是什么?
- Python 库,逐层加载大语言模型,让 70B 级模型可在单张 4GB 显卡上推理,无需量化。 AirLLM 是一个 Python 库,用于在显存很小的 GPU 上运行大语言模型。它不会一次性把整个模型装入显存,而是每次只在 GPU 上保留一个解码层,因此所需显存取决于单层大小,而不是模型总参数量。项目称可在约 4GB 显存上运行 Llama 3.x 70B,在约 12GB 上运行 DeepSeek-V3(671B),并支持 Qwen3-235B、Kimi K3 等新模型在单卡上推理。
- airllm 可以免费使用吗?用的什么许可证?
- airllm 以 Apache-2.0 许可证开源。用于商业产品前,请先确认许可证条款。
- airllm 还在维护吗?
- 活跃维护中。airllm 最近一次提交是在 2026年10月19日。
- airllm 适合用来做什么?
- 在显存有限的消费级显卡上体验 Llama 70B 等大型开源模型。 在单台工作站显卡上运行 DeepSeek-V3 这类超大 MoE 模型。 在小显存 GPU 上使用 LoRA 微调大模型。 在 Apple 芯片 Mac 上本地运行大模型。
- airllm 怎么安装和运行?
- 安装 airllm 包: pip install airllm 通过 Hugging Face 仓库 ID 加载模型并运行推理(参见 README 中的快速开始代码): model = AutoModel.from_pretrained("Qwen/Qwen3-32B")
- 在哪里获取 airllm?
- 源代码和安装说明在 GitHub 上:https://github.com/lyogavin/airllm
概要由 AI 根据项目 README 生成;GitHub 数据每日更新。
相关项目
在家用 NVIDIA 或 AMD 显卡上本地运行大参数 Qwen3.8-Flash-Next 模型的推理引擎,提供兼容 OpenAI 与 Anthropic 的本地 API。
利用导出的聊天记录微调大模型,生成模仿你说话风格的数字分身,并部署为聊天机器人。
面向消费级硬件的边缘端 MoE 推理引擎,可在个人电脑上运行大型开源权重模型。
面向 Stable Diffusion 等图像生成模型的训练、生成与工具脚本集合,支持 LoRA 与微调。
一款命令行工具,自动检测 GPU、CPU 和内存,从 HuggingFace 上挑选真正能在你的机器上运行、且表现最好的本地大模型。