qidao123.com技术社区-IT企服评测·应用市场

标题: DeepSeek模型微调：使用unsloth微调框架对DeepSeek-R1-Distill-Llama-8B模 [打印本页]

---

作者: 耶耶耶耶耶    时间: 2025-2-28 07:46
标题: DeepSeek模型微调：使用unsloth微调框架对DeepSeek-R1-Distill-Llama-8B模
本文介绍了使用unsloth微调框架对DeepSeek-R1-Distill-Llama-8B模型进行微调,实现将SQL语句转换为自然语言描述。主要步调包括:
1️⃣在Colab设置运行情况,安装必要的库和包
2️⃣预备和处理惩罚huggingface上的数据集
3️⃣设置微调的超参数,启动微调过程
4️⃣测试微调后模型的性能,生存并上传微调的模型
  

---

1️⃣ Colab情况设置与依赖安装

核心步调：

• 安装Unsloth库
  在Colab中通过以下命令安装Unsloth及其依赖，确保支持4bit量化和高效训练加速：
  
  1. !pip install unsloth
  2. !pip install --force-reinstall --no-cache-dir --no-deps git+https://github.com/unslothai/unsloth.git

  复制代码
  Unsloth通过Triton语言优化了模型训练内核，相比传统方法可减少70%显存占用并提速2倍34。
  
• 登录Hugging Face与Weights & Biases
  从Colab密钥获取Hugging Face和Wandb的访问令牌，用于数据集加载与实验跟踪：
  
  1. from huggingface_hub import login
  2. login(userdata.get('HF_TOKEN'))  # Hugging Face登录
  3. import wandb
  4. wandb.login(key=userdata.get('WB_TOKEN'))  # 实验跟踪

  复制代码

---

2️⃣ 数据集预备与预处理惩罚

关键要点：

• 数据集选择
  推荐使用Hugging Face上的b-mc2/sql-create-context数据集，该数据集包含SQL语句与对应的自然语言描述，实用于训练SQL转文本任务1。
  
• 数据格式转换
  将数据集处理惩罚为Unsloth支持的指令-响应格式，例如：
  
  1. def format_sql_to_text(example):
  2.     prompt = f"### SQL Query:\n{example['sql']}\n### Natural Language Description:"
  3.     response = example['description']
  4.     return {"text": f"{prompt}{response}"}
  5. dataset = dataset.map(format_sql_to_text)

  复制代码
  需确保输入包含SQL语句与目标描述的映射，并统一使用### Query:和### Response:作为分隔符16。
  

---

3️⃣ 模型加载与微调参数设置

优化设置：

• 加载DeepSeek-R1-Distill-Llama-8B模型
  使用4bit量化加载模型以低落显存需求，设置最大序列长度为2048：
  
  1. from unsloth import FastLanguageModel
  2. model, tokenizer = FastLanguageModel.from_pretrained(
  3.     model_name="unsloth/DeepSeek-R1-Distill-Llama-8B",
  4.     max_seq_length=2048,
  5.     load_in_4bit=True,
  6.     token=hf_token
  7. )

  复制代码
  此设置可在16GB显存的GPU（如T4）上运行15。
  
• LoRA微调参数
  添加低秩适配器（LoRA）进行高效微调，推荐参数：
  
  1. model = FastLanguageModel.get_peft_model(
  2.     model,
  3.     r=16,              # LoRA秩，平衡速度与精度
  4.     lora_alpha=32,     # 缩放因子，增强适配强度
  5.     target_modules=["q_proj", "k_proj", "v_proj", "o_proj"],
  6.     use_gradient_checkpointing="unsloth",  # 长序列优化
  7. )

  复制代码
  该设置在保持精度的同时减少约40%显存占用34。
  

---

4️⃣ 启动微调与训练监控

执行流程：

• 训练参数设置
  
  1. from transformers import TrainingArguments
  2. trainer = TrainingArguments(
  3.     per_device_train_batch_size=2,
  4.     gradient_accumulation_steps=4,
  5.     warmup_steps=10,
  6.     max_steps=100,
  7.     learning_rate=2e-4,
  8.     fp16=torch.cuda.is_bf16_supported(),
  9.     optim="adamw_8bit",
  10.     logging_steps=1,
  11.     report_to="wandb",  # 实时监控训练指标
  12. )

  复制代码
  发起使用Wandb跟踪丧失曲线和资源消耗56。
  
• 启动训练
  使用SFTTrainer加载数据并启动微调：
  
  1. from trl import SFTTrainer
  2. trainer = SFTTrainer(
  3.     model=model,
  4.     train_dataset=dataset,
  5.     dataset_text_field="text",
  6.     max_seq_length=2048,
  7.     tokenizer=tokenizer,
  8. )
  9. trainer.train()

  复制代码
  典型训练时间约20-30分钟（Colab T4情况）15。
  

---

5️⃣ 模型测试与部署

验证与导出：

• 性能测试
  使用推理代码验证模型天生效果：
  
  1. prompt = "### Query:\nSELECT customer_id, SUM(amount) FROM orders GROUP BY customer_id;\n### Response:"
  2. inputs = tokenizer([prompt], return_tensors="pt").to("cuda")
  3. outputs = model.generate(**inputs, max_new_tokens=200)
  4. print(tokenizer.decode(outputs[0]))

  复制代码
  微调后模型应能天生如“统计每位客户的总订单金额”的清楚描述16。
  
• 模型生存与上传
  当地生存适配器并推送至Hugging Face Hub：
  
  1. model.save_pretrained("sql_to_nl_adapter")
  2. model.push_to_hub("your-username/DeepSeek-R1-SQL2Text")
  3. tokenizer.push_to_hub("your-username/DeepSeek-R1-SQL2Text")

  复制代码
  支持后续部署至Ollama或API服务35。
  

---

优化发起与常见问题

• 显存不足处理惩罚：若遇到OOM错误，可实验低落per_device_train_batch_size或启用梯度查抄点4。
  
• 精度提升：增加LoRA的r值（如32）或使用全参数微调（需更高显存）3。
  
• 多任务扩展：结合医疗、金融等领域数据集，可进一步优化模型领域顺应性25。
  

---

links：
https://www.bilibili.com/video/BV1pCNaeaEEJ

免责声明：如果侵犯了您的权益，请联系站长，我们会及时删除侵权内容，谢谢合作！更多信息从访问主页：qidao123.com:ToB企服之家，中国第一个企服评测及商务社交产业平台。

---

欢迎光临 qidao123.com技术社区-IT企服评测·应用市场 (https://dis.qidao123.com/)

Powered by Discuz! X3.4