项目预览
项目数据
项目介绍
Star 增长趋势
本站每日记录该项目 GitHub 星标变化,观测越久曲线越完整
数据来源:GitHub API 逐日快照 · 收录于 2026-08-09 · 当前星标 ★ 4460
使用教程
环境要求
- 已安装 PyTorch、CMake 和 packaging,否则需先执行 pip install cmake packaging torch
- 推荐在 mosaicml/llm-foundry Docker 镜像内运行(如 2.7.0_cu128-latest)
- 有 NVIDIA GPU 时使用 pip install -e ".[gpu]",无 GPU 用 pip install -e .
- 不使用 Docker 时建议先创建并激活虚拟环境
- 官方 Docker 镜像基于 Python 3.12、Torch 2.7.0、CUDA 12.8
安装与启动步骤
-
1克隆代码仓库
从 GitHub 克隆 llm-foundry 仓库到本地,为后续安装和运行脚本做准备。
git clone https://github.com/mosaicml/llm-foundry.git -
2进入项目目录
切换到克隆下来的 llm-foundry 目录,后续安装命令都在此目录执行。
cd llm-foundry -
3安装基础依赖
README 假设已装 PyTorch、CMake、packaging;若没有,用此命令安装。
pip install cmake packaging torch -
4安装 GPU 版本
有 NVIDIA GPU 时安装带 gpu 可选依赖的 llm-foundry 包,使用可编辑模式。
pip install -e ".[gpu]" -
5安装 CPU 版本
若无 NVIDIA GPU,改用不带 gpu 可选依赖的可编辑安装。
pip install -e . -
6进入脚本目录
Quickstart 从 scripts 目录开始,可继续跑数据准备、训练与推理脚本。
cd scripts
如何确认成功
按 Quickstart 跑通 C4 数据准备、MPT-125M 训练、HuggingFace 导出和评估即成功。
常见问题
Q:Docker 镜像里已经装好 llm-foundry 了吗?
A:没有。mosaicml/llm-foundry 镜像只预装依赖,仍需用 pip install llm-foundry 或从源码 pip install -e 安装。
Q:没有 NVIDIA GPU 怎么安装?
A:使用 pip install -e . 安装,不要加 [gpu] 可选依赖即可。
Q:安装后依赖冲突怎么办?
A:README 提示可能需要升降级部分包,例如将 numpy 降到 1.23.5,再重试。
Q:Quickstart 只训练 10 个 batch 效果好吗?
A:只是演示工具流程,要获得好的模型质量需训练远超 10 个 batch。
注意事项
- 官方强烈推荐在 Docker 容器中使用 llm-foundry。
- Docker 镜像不包含 llm-foundry 包,只包含依赖。
- 不用 Docker 时建议创建虚拟环境;ROCm 用户还需安装 ROCm 版 flash attention。
- README 的 10 batch 训练仅用于演示,不要期望模型质量。
核心亮点
- 集成 Composer 和 FSDP,支持大规模分布式训练,性能优化成熟
- 内置 StreamingDataset,可高效处理 TB 级数据流,减少 I/O 瓶颈
- 提供从数据准备到评估的完整脚本,降低入门门槛,支持快速复现实验
不足之处
- 文档偏技术化,新手上手成本较高
- 对非 Databricks 环境的兼容性依赖较多,社区支持待观察
适用场景
- 企业级 LLM 预训练和持续预训练
- 指令微调与 RLHF 流程的快速落地
- 基于 Hugging Face 模型的自定义训练与评估
替代项目
NVIDIA NeMo、Hugging Face Transformers + Trainer、EleutherAI 的 GPT-NeoX
下一篇:SwanLab
同类项目推荐
residual-context-diffusion
开源
把丢弃的信号变成推理先验,让扩散大模型更聪明
[ICML 2026] Residual Context Diffusion (RCD): Repurposing discarded signals as struc···
PaddleFormers
开源
在飞桨上跑通大模型微调与训练,开箱即用
PaddleFormers is an easy-to-use library of pre-trained large language model zoo base···
Agent-R1
开源
端到端强化学习,训练出能自主决策的 LLM Agent
Agent-R1: Training Powerful LLM Agents with End-to-End Reinforcement Learning
trl
开源
几行代码给大模型做 RLHF 对齐与偏好微调
Train transformer language models with reinforcement learning.
spirit-lora-trainer
开源
拖拽配置,一键训练 Flux1 LoRA,告别命令行烦恼
Spirit Lora Trainer is a robust toolkit for training Flux1-LoRA models with a focus ···
Spatial-SSRL
开源
让AI看懂空间关系,无需人工标注,强化学习自动提升空间推理。
[CVPR 2026] Official release of "Spatial-SSRL: Enhancing Spatial Understanding via S···
EvoControl
开源
让AI自己优化代码,还能按你的要求来
Official implementation of Controlled Self-Evolution for Algorithmic Code Optimizati···
paradigms
开源
并行采样加速扩散模型,少步生成高质量图像
PyTorch implementation for "Parallel Sampling of Diffusion Models", NeurIPS 2023 Spo···
DMax
开源
让扩散大模型生成提速,并行解码不再卡顿
[NeurIPS 2026] DMax: Aggressive Parallel Decoding for dLLMs
