保姆级教程:零代码基础也能微调Qwen3,并本地部署

我将在本文介绍如何通过 unsloth 框架以 LoRA 的方法微调 Qwen3-14B 模型。
到目前还有很多小伙伴还不明白什么时候应该微调?那么请看下图:
接下来我们再看一下本文使用的 LoRA 微调方法的优势:

LoRA(Low-Rank Adaptation of Large Language Models,大型语言模型的低秩自适应)是一种流行的轻量级训练技术,可以显著减少可训练参数的数量。它的工作原理是将少量的新权重插入模型中,并且只训练这些权重。这使得使用 LoRA 进行训练的速度更快、内存效率更高,并且生成的模型权重更小(只有几百 MB),更易于存储和共享。LoRA 还可以与 DreamBooth 等其他训练技术结合使用,以加速训练。

我们将在本文介绍如何微调使模型成为一个”双重人格”的助手,既能进行普通闲聊,又能在需要时切换到更严谨的思考模式来解决复杂问题,特别是数学问题。简而言之,微调后的模型获得的能力:

1. 双模式操作能力:

  • 普通对话模式: 适用于日常聊天场景。

  • 思考模式( Thinking Mode ): 用于解决需要推理的问题。

2. 数学推理能力: 能够解决数学问题并展示详细的推理过程,如示例中的”解方程(x + 2)^2 = 0″。

3. 对话能力保持: 同时保持了自然对话的能力,能够进行流畅的多轮对话。

首先我们在谷歌 Colab 上选择算力,推荐使用 T4 GPU 或者 A100 GPU:

现在我们可以加载 14 B模型:

我们现在添加 LoRA 适配器,因此我们只需要更新 1% 到 10% 的参数!

准备数据

Qwen3 既有推理模式,也有非推理模式。因此,我们应该使用两个数据集:

  • Open Math Reasoning 数据集,该数据集曾用于赢得 AIMO(AI Mathematical Olympiad,AI 数学奥林匹克 – 进步奖 2)挑战赛!我们从使用 DeepSeek R1 的可验证推理轨迹中抽取了 10%,其准确率超过 95%。

  • 我们还利用了 Maxime Labonne 的 FineTome-100k 数据集(ShareGPT 格式)。但我们还需要将其转换为 HuggingFace 的常规多轮对话格式。

我们现在将推理数据集转换为对话格式:

接下来,我们将非推理数据集也转换为对话格式。

首先,我们必须使用 Unsloth 的 standardize_sharegpt 函数来修复数据集的格式。

非推理数据集要长得多。假设我们希望模型保留一些推理能力,但我们特别想要一个聊天模型。

让我们定义一个纯聊天数据的比例。目标是定义两种数据集的某种混合。让我们选择 25% 的推理数据和 75% 的聊天数据:

最后合并数据集:

训练模型

现在让我们使用 Huggingface TRL 的 SFTTrainer!我们执行 60 步来加快速度,但你可以设置 num_train_epochs=1 进行完整运行,并关闭 max_steps=None。

让我们开始训练模型吧!要恢复训练,请设置 trainer.train(resume_from_checkpoint = True)

推理

让我们通过 Unsloth 原生推理来运行模型!根据 Qwen-3 团队的说法,

  • 推理的推荐设置是:temperature = 0.6、top_p = 0.95、top_k = 20。

  • 对于基于普通聊天的推理,temperature = 0.7、top_p = 0.8、top_k = 20。

保存、加载微调模型

要将最终模型保存为 LoRA 适配器,请使用 Huggingface 的 push_to_hub 进行在线保存,或使用 save_pretrained 进行本地保存。

[注意] 这仅保存 LoRA 适配器,而不是完整模型。后面我来介绍如何保存为 16 位或 GGUF 格式。

现在,如果你想加载我们刚刚保存用于推理的 LoRA 适配器,请将 False 设置为 True:

保存为 VLLM 的 float16

选择 merged_16bit 保存 float16,或选择 merged_4bit 保存 int4。使用 push_to_hub_merged 上传到你个人的 Hugging Face 账户!

GGUF / llama.cpp 转换

使用 save_pretrained_gguf 进行本地保存,使用 push_to_hub_gguf 上传到 HF。

  • q8_0 – 快速转换。资源占用较高,但通常可以接受。

  • q4_k_m – 推荐。使用 Q6_K 处理 attention.wv 和 feed_forward.w2 张量的一半,否则使用 Q4_K。

  • q5_k_m – 推荐。使用 Q6_K 处理 attention.wv 和 feed_forward.w2 张量的一半,否则使用 Q5_K。

本地部署

接下来就是将 GGUF 文件下载到本地,以便本地部署运行。

加入星球获取精心打造的提示工程RAGAgent开发实践教程。

https://colab.research.google.com/drive/1pfv8UvNyLhB0ZKbhl7jbCeHcHBUN12Ai?usp=sharing#scrollTo=iX3Hv8YzBKyr

(文:PyTorch研习社)

发表评论

×

下载每时AI手机APP

 

和大家一起交流AI最新资讯!

立即前往