
LlamaFactory
上百种模型随意挑,教出你专属的 AI
项目预览
项目数据
项目介绍
Star 增长趋势
本站每日记录该项目 GitHub 星标变化,观测越久曲线越完整
数据来源:GitHub API 逐日快照 · 收录于 2026-08-09 · 当前星标 ★ 75956
技术标签
使用教程
环境要求
- Python 3.11 环境(官方 Docker 镜像内置 PyTorch 2.6.0;Ascend NPU 需 Python 3.12)
- NVIDIA GPU 及可用 CUDA(Docker 方式以 --gpus=all 调用)
- 已安装 Git,用于克隆源码
- Windows 需手动安装 GPU 版 PyTorch;使用 uv 方式需先安装 uv
安装与启动步骤
-
1克隆仓库
下载 LlamaFactory 源码并进入目录,--depth 1 只拉取最新一次提交。
git clone --depth 1 https://github.com/hiyouga/LlamaFactory.git cd LlamaFactory -
2装 Windows PyTorch
仅 Windows 用户需手动安装 GPU 版 PyTorch,装完确认 CUDA 可用再继续。
pip uninstall torch torchvision torchaudio pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu126 python -c "import torch; print(torch.cuda.is_available())" -
3安装项目依赖
在源码目录内以可编辑模式安装本体,并补充 metrics 指标依赖。
pip install -e . pip install -r requirements/metrics.txt -
4安装可选依赖
需要 DeepSpeed 等可选能力时,一次性安装 metrics 与 deepspeed 依赖。
pip install -e . && pip install -r requirements/metrics.txt -r requirements/deepspeed.txt -
5Docker 方式运行
不想配环境可直接拉官方镜像进入容器,需宿主机支持 GPU 容器运行时。
docker run -it --rm --gpus=all --ipc=host hiyouga/llamafactory:latest -
6启动 Web UI
用 uv 创建隔离环境并启动 Gradio 网页界面,实现零代码微调。
uv run llamafactory-cli webui -
7运行 LoRA 微调
以 Qwen3-4B-Instruct 为例执行 LoRA 微调,配置路径相对仓库根目录。
llamafactory-cli train examples/train_lora/qwen3_lora_sft.yaml -
8推理与合并模型
先加载微调结果对话验证效果,再导出合并后的完整模型权重。
llamafactory-cli chat examples/inference/qwen3_lora_sft.yaml llamafactory-cli export examples/merge_lora/qwen3_lora_sft.yaml
关键配置
| 配置项 | 必填 | 说明 | 示例 |
|---|---|---|---|
dataloader_num_workers | 否 | Windows 出现 Can't pickle local object 报错时设为 0 | 0 |
double_quantization | 否 | Ascend NPU 上使用 QLoRA 时需关闭双量化 | false |
如何确认成功
执行 llamafactory-cli help 能显示帮助信息;Windows 下 python -c "import torch; print(torch.cuda.is_available())" 输出 True。
常见问题
Q:Windows 训练时报 Can't pickle local object 怎么办?
A:在配置中尝试设置 dataloader_num_workers: 0,减少多进程数据加载带来的序列化问题。
Q:Windows 上想用 QLoRA 量化微调?
A:需要额外安装 bitsandbytes:pip install bitsandbytes;使用 uv 环境时先装 GPU 版 PyTorch,再执行 uv pip install bitsandbytes --no-deps。
Q:Ascend NPU 怎么安装?
A:使用 Python 3.12,执行 pip install -r requirements/npu.txt,并安装 Ascend CANN Toolkit 与 Kernels,也可直接拉取预构建的 NPU Docker 镜像。
Q:不想本地装依赖怎么最快跑起来?
A:直接用官方镜像:docker run -it --rm --gpus=all --ipc=host hiyouga/llamafactory:latest,或使用 README 中的 Colab、PAI-DSW 等云端环境。
注意事项
- 安装是必须步骤,README 明确标注 Installation is mandatory。
- 特定功能的额外依赖放在 examples/requirements/ 目录下,按需安装。
- Windows 安装 bitsandbytes 要注意 CUDA Toolkit 版本,RTX 50 系列需 CUDA 12.8–12.9 的构建。
- 除 README 列出的链接外,其他网站均为未授权第三方网站,请谨慎使用。
核心亮点
- 统一高效微调框架,支持100+大语言模型与视觉语言模型,覆盖LoRA、QLoRA、MoE等多种微调策略,适配性强
- 集成DeepSpeed、FlashAttention等优化技术,训练效率高,且提供极简的网页UI和命令行接口,降低使用门槛
- 基于ACL 2024论文,学术背书扎实,社区活跃(73.9k星标),持续更新迭代,生态成熟
不足之处
- 文档/社区待观察
- 对新手而言,配置多模型多后端的环境依赖较复杂,需一定深度学习基础
适用场景
- 学术研究:快速复现和对比不同微调方法在特定任务上的效果
- 企业私有化部署:针对行业数据微调开源LLM以提升垂直领域性能
- 个人开发者:低成本微调小参数模型用于智能助手或内容生成应用
替代项目
Hugging Face PEFT、Unsloth、Axolotl
下一篇:minimind
同类项目推荐
Model-Optimizer
开源
一站式压缩大模型,导出即加速推理
A unified library of SOTA model optimization techniques like quantization, distillat···
Time-R1
开源
用三阶段强化学习,让大模型真正理解时间、预测未来。
Time-R1: Framework and resources for endowing LLMs with comprehensive temporal reaso···
selfcodealign
开源
让代码模型自己教自己,无需人工标注,自动提升生成质量。
[NeurIPS'24] SelfCodeAlign: Self-Alignment for Code Generation
higgsfield
开源
让万亿参数模型训练不再怕GPU故障,自动容错编排
Fault-tolerant, highly scalable GPU orchestration, and a machine learning framework ···
RandOpt
开源
在预训练权重附近挖出多个专家模型,提升泛化与集成效果
Official Codebase for "Neural Thickets: Diverse Task Experts Are Dense Around Pretra···
pytorch-generative
开源
一键复现经典生成模型,快速上手图像生成实验
Easy generative modeling in PyTorch
AgileRL
开源
让强化学习训练快 10 倍,超参自动调优,开箱即用
The complete post-training stack for LLMs, plus RL for any problem. SFT, DPO and mul···
neural-compressor
开源
一键压缩大模型,低比特量化提速省内存
SOTA low-bit LLM quantization (INT8/FP8/MXFP8/INT4/MXFP4/NVFP4) & sparsity; leading ···
diffusers-torchao
开源
一键优化扩散模型,推理提速3倍,显存减半
End-to-end recipes for optimizing diffusion models with torchao and diffusers (infer···