notebooks
超250个即开即用的Jupyter Notebook,覆盖Llama/Qwen/Gemma等主流大模型的一键微调流程
加载项目详情…
本应用为开源项目,仅供学习研究,请遵守其开源协议。
超250个即开即用的Jupyter Notebook,覆盖Llama/Qwen/Gemma等主流大模型的一键微调流程
加载项目详情…
本应用为开源项目,仅供学习研究,请遵守其开源协议。
你有没有过这种感觉——花了大价钱调用闭源大模型的 API,却总觉得它「不懂」你的业务?它知道天下大事,却不知道你们公司内部怎么称呼客户;它能写漂亮的营销文案,却学不会你们团队的行话和语气。问题不在模型本身,而在于模型没有经过「定制化训练」。
unslothai/notebooks 就是来解决这个问题的。它不是一个模型,而是一套 250 多个可运行的 Jupyter Notebook,每个 notebook 对应一种模型、一种硬件平台、一种微调方法。你要微调哪个模型,就在对应的 notebook 里填入自己的数据集,按下运行键——Colab 会帮你搞定 GPU 计算,几小时后,一个「懂你」的大模型就训练完成了。
Unsloth 由 Daniel Han 和 Michael Han 创立于 2023 年,是一家专注于 LLM 训练效率优化的开源公司。团队的核心产品是 unsloth Python 库,该库通过动态量化、梯度检查点、PEFT(Parameter-Efficient Fine-Tuning)等多种技术,将大模型微调的显存占用降低至原来的 60-70%,同时将训练速度提升 2-5 倍。这意味着在一块消费级 RTX 4090 显卡上,就能微调一个 70 亿参数的大模型,而不再必须依赖昂贵的 A100/H100 集群。
notebooks 仓库正是 Unsloth 团队将这些能力落地的「用户界面」。每个 notebook 预先配置好 unsloth 库、transformers、trl 等依赖,用户无需编写底层训练代码,只需准备数据集并修改超参数。这大幅降低了 AI 定制化训练的门槛,让没有专业 ML 工程团队的中小企业、独立开发者、甚至高校研究者,都能拥有自己的专属大模型。
该仓库的 notebook 覆盖了目前最主流的 100 多个大语言模型和多模态模型,类型包括:
| 类别 | 代表模型 |
|---|---|
| 文本生成(Text) | Llama 3/3.1/3.2, Mistral, Gemma 2/3, Phi-4, Qwen 2.5/3, DeepSeek R1, WizardLM, Yi |
| 视觉语言(Vision) | LLaVA, Qwen2-VL, Gemma 4 Vision, DeepSeek-VL, InternVL3 |
| 嵌入模型(Embedding) | BGE-M3, Nomic-Embed, GTE, All-MiniLM |
| 语音合成(TTS) | Bark, Parler-TTS, CosyVoice |
| 推理增强(Reasoning) | DeepSeek R1(GRPO), QwQ(思维链) |
对于每类模型,仓库还区分了不同参数规模(如 2B、3B、7B、8B、14B、70B 等),用户可以根据自己的硬件条件选择最合适的 notebook。
不同的训练目标需要不同的微调方法。仓库对主流方法均有覆盖:
仓库针对不同用户群体提供了多种运行方式:
colab-*.ipynb 文件,点击即运行,适合快速体验和轻量实验AMD-*.ipynb),包括 RDNA3/RDNA4 系列显卡,让 AMD 用户也能用上高效微调Dockerfile_DGX_Spark,面向企业用户,支持在 NVIDIA DGX 或 Spark 集群上部署unsloth_windows.ps1,简化 Windows 环境的安装配置仓库还包含多个维护工具脚本,展现了工程上的完整性:
update_all_notebooks.py(25.7 万字节):批量更新所有 notebook 中的依赖版本、代码模板,确保与最新库版本兼容update_max_seq_length.py:批量修改 notebook 的最大序列长度参数replace_text.py:批量替换 notebook 中的文本内容molab/ 目录:提供交互式 Web UI,辅助生成和编辑 notebook从代码结构上看,这是一个以 Notebook 为核心的分布式教程库,而非传统意义上的代码仓库。
技术栈:
架构特点:
pyproject.toml 管理项目依赖,用 uv.lock 锁定版本scripts/ 和 python_scripts/ 目录提供自动化脚本,支持 CI/CD 化的 notebook 生成tests/ 目录包含 pytest 测试,用 marimo 框架做 notebook 有效性验证代码质量评分:该仓库的代码质量较高,体现在:每个 notebook 都经过工程化脚本批量验证;提供了完整的 Docker 部署方案;文档详尽(README 长达 15.7 万字节);依赖管理规范(pyproject.toml + uv.lock)。缺点是没有传统意义上的单元测试覆盖(tests/ 主要验证 notebook 结构而非模型训练效果)。
对于大多数用户来说,上手路径非常简单:
方式一:Colab(最简单)
方式二:本地 Jupyter
git clone https://github.com/unslothai/notebookspip install unsloth unsloth_zoojupyter notebook显存需求估算:使用 QLoRA 微调,8B 参数模型最低只需 8GB 显存(量化后),70B 参数模型需要 40GB+。Colab 免费版提供 15GB 显存的 T4 显卡,可完整运行 7B 模型的微调。
这些 notebook 是教学和实验导向的,每个 notebook 的训练参数都是「开箱即用」的固定配置。如果你要做超大规模预训练(如训练一个全新基座模型)、多节点分布式训练、或需要精确的学习率调度,需要在此基础上做大量二次开发。notebook 模式天然不适合长时间、大规模的生产训练任务。
unshoth 的核心优势建立在对 transformers、bitsandbytes、triton 等库的 hack 级别的优化上。一旦上游库发布重大更新,notebook 中的代码可能需要等待 unsloth 团队同步更新。从 update_all_notebooks.py 脚本的体积(25.7 万字节)可以看出,团队在维护兼容性上投入了大量精力。
pyproject.toml 要求 Python >= 3.13,而 3.13 目前仍是较新的版本,部分第三方库可能尚未完全兼容。选择这个版本可能是为了利用 3.13 的新特性,但也意味着用户需要面对潜在的兼容性问题。
unslothai/notebooks 的出现,反映了 LLM 落地领域的一个大趋势:从「云端大模型 API」到「本地定制小模型」的范式转移。
过去两年,Mistral、Llama、Qwen 等开源模型的能力快速追赶 GPT-4,与此同时,Google Colab 的免费 T4 显卡(15GB 显存)让普通用户也能微调 7B 模型。Unsloth 通过 QLoRA 技术进一步压低硬件门槛——一块 RTX 4060(8GB 显存)就能微调 Llama 3 8B 版本。这不是小众极客的需求,而是大量中小企业的真实诉求:一个了解行业术语、能处理内部数据、调用成本接近零的专属 AI 助手。
从 Stars 增长曲线看,2024 年初该项目约 2000 星,到 2026 年已达 5300+ 星,增长超过 2.5 倍。这个数字背后不是极客的好奇心,而是企业级需求的真实积累。每一个新增的 notebook,都对应着某个团队在某个场景下的定制化训练尝试。
如果你的团队正在考虑用大模型改造业务流程,unslothai/notebooks 仓库是值得收藏的第一站。