← 返回 AI 开源导航

Unsloth

训练与微调协议 Apache-2.0★ 78.6kPython

一、项目简介

Unsloth 是高效的大模型微调框架。它通过手写计算内核与显存优化,让 LoRA / QLoRA 微调速度提升 2–5 倍、显存占用显著下降,使消费级显卡也能高效微调较大模型。

二、核心特性

  • 训练提速:相比原生实现快 2–5 倍;
  • 显存优化:大幅降低微调显存门槛;
  • 主流模型支持:Llama、Qwen、Mistral、Gemma 等;
  • 兼容 HF 生态:可导出合并后的模型;
  • 支持 LoRA / QLoRA;
  • Notebook 友好:示例流程清晰。

三、适用场景

  • 显存受限下的模型微调;
  • 需要快速迭代实验的团队;
  • 个人/小团队定制专属模型。

四、快速上手

pip install unsloth
from unsloth import FastLanguageModel
model, tok = FastLanguageModel.from_pretrained("Qwen/Qwen2.5-7B-Instruct")
# 配置 LoRA 后即可训练

五、注意事项

  • 主要支持特定模型与 NVIDIA GPU;
  • 需关注版本与模型兼容性;
  • 训练超参与数据质量仍是效果关键。