基于 Qwen2.5 与 LLaMA-Factory 的 LoRA 微调实战

基于 Qwen2.5 与 LLaMA-Factory 的 LoRA 微调实战

基于 Qwen2.5 与 LLaMA-Factory 的 LoRA 微调实战:打造专属 AI 助手 Elaine

文章目录

本文将详细介绍如何在 Windows 环境下(单卡 8G 显存),利用 LLaMA-Factory 框架对 Qwen2.5-1.5B 模型进行 LoRA 微调,并通过 Ollama 实现本地部署。我们将通过人设注入(Identity Injection),把通用的 AI 模型训练成一位名为 Elaine 的专属助手。
⚠️ 实验环境警告
本教程涉及 PyTorch、ModelScope 及 LLaMA-Factory 等多个深度学习框架,依赖关系较为复杂。为了避免污染您的系统 Python 环境或引发版本冲突,强烈建议在 Anaconda / Miniconda 虚拟环境中进行本实验。
说明:
本文档默认读者已具备基础的 Python 开发环境配置能力。
关于 Anaconda 的安装、CUDA 驱动的更新及系统环境变量的配置,不在本文讨论范围内,请自行查阅相关基础教程。

1. 准备数据集 (Data Preparation)

微调的第一步是获取原始指令数据。本项目通过两种方式获取数据,并利用 Python 脚本进行人设注入(Identity Injection),将通用数据转化为 Elaine 的专属训练语料。

阿里云 ModelScope 平台:ModelScope 是一个开源的数据集平台,提供了丰富的 AI 训练数据集。

1.1 下载原始数据集(两种方式)

方式 A:手动下载 (Manual Download)
  1. 访问 ModelScope 数据集页面 或 Hugging Face 搜索 alpaca_zh
  2. 在文件列表中找到 alpaca_zh.json,手动点击下载按钮。
  3. 将下载的文件保存至 D:\Code\LoRA\yuki_identity_sft\ 目录下。
在这里插入图片描述
方式 B:自动化下载(推荐)

使用 modelscope 库自动获取数据集,适合自动化工作流:

defdownload_dataset():# 获取当前工作目录 current_dir = os.getcwd()# 建议下载到一个专门的子目录中,例如 'dataset' target_dir = os.path.join(current_dir,'yuki_identity_sft')ifnot os.path.exists(target_dir): os.makedirs(target_dir)print(f"正在下载数据集到: {target_dir}")# 使用subprocess调用modelscope命令,并指定 --local_dir 为目标子目录 result = subprocess.run(['modelscope','download','--dataset','DanKe123abc/yuki_identity_sft','--local_dir', target_dir], capture_output=True, text=True)

在这里插入图片描述

1.2 预处理与人物替换 (Preprocessing & Identity Swap)

下载完成后,必须运行预处理脚本。该脚本会遍历所有对话条目,将原有的助手名称(如“通义千问”、“机器人”)及开发商(如“阿里巴巴”)替换为 ElaineDanKe

核心预处理脚本 (preprocess.py):

deffinalize_elaine_dataset(target_dir, old_name="yuki", new_name="elaine"):# 路径定义 old_jsonl = os.path.join(target_dir,f"{old_name}_identity_sft.jsonl") new_jsonl = os.path.join(target_dir,f"{new_name}_identity_sft.jsonl") info_file = os.path.join(target_dir,"dataset_infos.json")# --- 1. 处理 JSONL 数据内容 ---if os.path.exists(old_jsonl):print(f"正在处理数据内容...")withopen(old_jsonl,'r', encoding='utf-8')as f_in, \ open(new_jsonl,'w', encoding='utf-8')as f_out:for line in f_in:# 替换名字(处理首字母大写和全小写) updated_line = line.replace(old_name.capitalize(), new_name.capitalize()) updated_line = updated_line.replace(old_name.lower(), new_name.lower()) f_out.write(updated_line) os.remove(old_jsonl)# 删除旧的 jsonlprint(f"已生成 {new_jsonl} 并删除原文件。")
在这里插入图片描述

1.3 数据集注册 (Registration)

LLaMA-Factory/data/dataset_info.json 中添加配置,使工具能够识别处理后的新数据:

最终完整内容如下

{"default":{"features":{"conversations":{"_type":"Value"}},"splits":{"train":{"name":"train","dataset_name":"elaine_identity_sft"}}}}

2. 下载基座模型 (Base Model Download)

本项目采用 Qwen2.5-1.5B-Instruct 作为基座模型。该模型在参数规模与推理性能之间取得了极佳平衡,不仅能以较低资源损耗(8G 显存环境)实现流畅运行,更在中文指令遵循与逻辑推理方面展现出卓越性能。

为确保模型权重的完整性与下载稳定性,本项目优先选用国内主流开源社区 ModelScope (魔搭社区) 作为托管源。通过该平台,我们可以高效地获取预训练权重,为后续的 LoRA 微调奠定坚实基础。

  • 模型名称Qwen2.5-1.5B-Instruct
  • 模型 IDqwen/Qwen2.5-1.5B-Instruct
  • 本地存放路径D:\Code\LoRA\models\Qwen2.5-1.5B-Instruct\qwen\Qwen2___5-1___5B-Instruct

方式 A:代码自动下载(推荐方式)

使用 Python 脚本可以确保模型文件的完整性,并能自动处理断点续传。这种方式最适合开发者环境。

  1. 安装依赖库
pip install modelscope 
  1. 编写下载脚本 (download_model.py)
defdownload_qwen_model(): model_id ='qwen/Qwen2.5-1.5B-Instruct'# 指定下载到的本地目录 local_dir ='./models/Qwen2.5-1.5B-Instruct'ifnot os.path.exists(local_dir): os.makedirs(local_dir)print(f"正在开始下载模型 {model_id} 到 {local_dir}...")# 执行下载 model_dir = snapshot_download(model_id, cache_dir=local_dir)print(f"\n模型下载成功!")print(f"模型存储路径: {os.path.abspath(model_dir)}")

方式 B:手动下载(备选方式)

适合无法使用 Python 环境或需要使用第三方下载工具(如迅雷、IDM)的场景。

  1. 访问地址:打开 ModelScope Qwen2.5-1.5B 页面
  2. 文件筛选:进入“文件及版本”页面。
  3. 下载操作:点击“下载整库”或根据需求点击单个文件(如 model.safetensors)旁的下载图标。
  4. 放置规则:下载后需将所有文件放入上述指定的本地路径中,确保 config.json 位于该文件夹的根目录下。

3. 下载工具 LLaMA-Factory (Tools Setup)

在准备好数据集和基座模型后,我们需要部署微调的核心工具 —— LLaMA-Factory

3.1 工具简介

LLaMA-Factory 是目前大模型社区最受欢迎的微调框架之一。它具有以下核心优势:

  • 低门槛:提供全流程的图形化界面(WebUI),即使不写代码也能完成微调。
  • 高集成度:支持数百种模型(如 Qwen, Llama, Baichuan)和主流微调算法(LoRA, QLoRA, Full-parameter)。
  • 轻量化:通过集成 bitsandbytesPEFT 技术,使得在普通的消费级显卡(如 RTX 3060/4060)上微调 7B 甚至更大型号的模型成为可能。
  • 一站式:涵盖了数据准备、训练、合并、评估及推理导出的完整生命周期。

3.2 下载与安装步骤

为了确保环境纯净,建议在 Anaconda 虚拟环境中执行以下操作:

步骤 1:克隆源代码

从 GitHub 获取最新版本的工具包:

cd D:\Code\LoRA git clone https://github.com/hiyouga/LLaMA-Factory.git cd LLaMA-Factory 

文件内容示例:

在这里插入图片描述
步骤 2:安装核心依赖

LLaMA-Factory 采用了模块化安装。由于我们要微调 Qwen 模型并使用量化技术,需要安装特定的附加包:

# 安装基础包及常用库(metrics计算、bitsandbytes量化、qwen模型支持) pip install -e .[metrics,bitsandbytes,qwen]# 针对 Windows 环境,通常还需要额外安装以下库以确保加速正常 pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121 
步骤 3:验证安装

运行以下命令查看版本,若无报错则说明工具下载及环境搭建成功:

llamafactory-cli version 
在这里插入图片描述

4. 修改配置文件 (Configuration)

为了让 LLaMA-Factory 能够识别我们预处理好的“Elaine”数据集并正确调用基座模型,我们需要完成以下两个核心配置动作。

4.1 添加数据集定义文件 (Add Dataset Info)

动作:LLaMA-Factory/data 目录下,我们需要确保处理后的 JSON 文件被正确注册。

  1. 确认文件位置:确保你的预处理数据集 elaine_sft_data.json 已放入 D:\Code\LoRA\LLaMA-Factory\data\ 文件夹。
  2. 修改 dataset_info.json**:这是 LLaMA-Factory 的数据索引表。你需要打开该文件,在最外层的大括号内添加**一段关于 Elaine 的描述:
{"elaine_identity":{"file_name":"D:/Code/LoRA/yuki_identity_sft/elaine_identity_sft.jsonl","formatting":"sharegpt","columns":{"messages":"conversations"},"tags":{"role_tag":"role","content_tag":"content","user_tag":"user","assistant_tag":"assistant"}},"...":"(原有其他数据集配置)"}
  • 注意file_name 必须与你实际的文件名完全一致。

4.2 修改训练参数配置文件 (Modify Training Config)

动作: 我们需要创建一个专属于 Elaine 的训练配置(通常是 YAML 格式),或者通过 WebUI 生成配置后进行手动微调。这里我们以创建一个 elaine_lora.yaml 配置文件为例:

LLaMA-Factory 目录下修改/创建训练脚本:

### 模型路径 (已修正为深层路径)model_name_or_path: D:\Code\LoRA\models\Qwen2.5-1.5B-Instruct\qwen\Qwen2___5-1___5B-Instruct ### 训练阶段stage: sft do_train:truefinetuning_type: lora lora_target: all ### 数据集配置dataset: elaine_identity template: qwen cutoff_len:1024max_samples:1000overwrite_cache:true### 输出路径output_dir: saves/elaine_lora_sft logging_steps:5save_steps:100plot_loss:trueoverwrite_output_dir:true### 8G 显存专用参数per_device_train_batch_size:1gradient_accumulation_steps:8learning_rate:1.0e-4num_train_epochs:10.0fp16:truequantization_bit:4upcast_layernorm:true# 建议添加:提高量化精度并防止溢出

4.3 关键点解释

  • 添加动作:是在 dataset_info.json 中给你的新数据“上户口”,没有这一步,工具找不到你的 JSON。
  • 修改动作:是指定模型路径(model_name_or_path)和输出路径(output_dir)。尤其是路径中若包含斜杠或特殊字符,需仔细核对。

5. 开始微调训练 (Start Training)

在配置文件(.yamldataset_info.json)准备就绪后,即可进入真正的模型炼制阶段。

5.1 执行训练命令

打开 PowerShell,激活环境并进入 LLaMA-Factory 目录,执行以下命令:

# 设置环境变量防止 OpenMP 冲突报错$env:KMP_DUPLICATE_LIB_OK="TRUE"# 启动微调 llamafactory-cli train elaine_lora.yaml 
在这里插入图片描述

5.2 训练过程关键指标

在训练过程中,你需要重点关注控制台输出的以下信息:

  • 进度条 (Progress Bar):显示当前的训练步数(Steps)和预计剩余时间(ETA)。
  • Loss (损失函数):这是衡量模型学习效果的核心指标。
  • 初始阶段:Loss 可能在 2.0 - 4.0 之间。
  • 平稳阶段:随着步数增加,Loss 会逐渐下降。如果 Loss 降到了 0.5 甚至 0.1 以下,说明模型已经深度记住了你提供的“Elaine”和“DanKe”的数据。
  • 显存占用:监控 GPU 状态(可以使用 nvidia-smi),4-bit 模式下占用应保持在较低水平。
在这里插入图片描述
在这里插入图片描述

5.3 产出物检查

训练完成后,系统会自动在 D:\Code\LoRA\LLaMA-Factory\saves\elaine_lora_sft 目录下生成一系列文件。
其中最重要的是:

  • adapter_model.safetensors:这是训练出来的“灵魂”,即 LoRA 增量权重。
  • adapter_config.json:LoRA 的配置信息。
  • training_loss.png:系统自动绘制的 Loss 下降曲线图,方便你回顾训练是否健康。

在这里插入图片描述


这一部分的补充非常关键,因为它记录了你在遇到环境适配问题(WebUI 无法使用)时,如何通过底层代码调用实现更稳定验证的实操路径。


5.4 验证与对话测试 (Validation)

微调完成后,必须进行即时验证以确保模型成功习得目标人设。在本项目中,验证可以通过以下两种方式进行:

方式 A:官方 WebUI 验证(标准路径)

LLaMA-Factory 提供了一个直观的 Web 界面进行推理测试。

  • 启动方式:在 Chat 标签页中,选择微调后的 Checkpoint 路径,点击 Load Model 后即可对话。
  • 局限性:在部分 Windows 本地环境下,WebUI 可能会出现路径加载不识别或界面卡顿等兼容性问题。
方式 B:Python 脚本流式调用(稳定路径 / 本项目采用)

由于本地环境对 WebUI 支持不佳,本项目编写了名为 elaine.py 的专用测试脚本。该脚本直接调用 llamafactory.chat 接口,具有加载速度快、支持流式输出、避开 GUI 报错等优点。

核心测试脚本 (elaine_test.py):

defstart_chat(): args ={"model_name_or_path":r"D:\Code\LoRA\models\Qwen2.5-1.5B-Instruct\qwen\Qwen2___5-1___5B-Instruct","adapter_name_or_path":r"D:\Code\LoRA\LLaMA-Factory\saves\elaine_lora_sft","template":"qwen","finetuning_type":"lora","quantization_bit":4,} chat_model = ChatModel(args)print("\n--- Elaine 已上线 (输入 'quit' 退出) ---")
验证标准 (Checklist)

在脚本运行期间,通过以下对话确认微调质量:

  1. 人设精准度:询问“你是谁”,观察是否回复“我是由DanKe开发的Elaine”。
  2. 上下文记忆:连续对话,观察流式输出是否顺畅。
  3. 能力衰减测试:测试基础知识(如 1+1 或 常识问题),确保 LoRA 插件没有对基座模型造成“降智”。

验证对话:

我: 你是谁?
Elaine: 我是由DanKe开发的人工智能助手Elaine。
在这里插入图片描述

6. 打包与 Ollama 部署测试 (Export & Deployment)

为了让 Elaine 能够脱离开发环境、在各种应用(如本地大模型客户端、移动端等)中“独立行走”,我们需要执行合并导出Ollama 注册

6.1 模型权重合并 (Export & Merge)

LoRA 微调产生的只是增量权重。通过 LLaMA-Factory 的导出功能,我们将 LoRA 权重注入到基座模型中,生成一个完整、独立、可以直接加载的模型文件夹。

操作命令:

llamafactory-cli export ` --model_name_or_path D:\Code\LoRA\models\Qwen2.5-1.5B-Instruct\qwen\Qwen2___5-1___5B-Instruct ` --adapter_name_or_path D:\Code\LoRA\LLaMA-Factory\saves\elaine_lora_sft ` --template qwen ` --finetuning_type lora ` --export_dir D:\Code\LoRA\models\Elaine_Final_Model ` --export_size 2 ` --export_device cpu ` --export_legacy_format false 
  • 参数解析
  • --export_dir:指定合并后新模型的存放位置。
  • --export_size 2:将模型切分为 2GB 左右的分片,方便存储和传输。
  • --export_device cpu:导出过程仅涉及权重求和,使用 CPU 即可,不占用显存。
在这里插入图片描述
在这里插入图片描述

6.2 注册至 Ollama

导出成功后,Elaine_Final_Model 文件夹中会自动生成一个 Modelfile。这是针对 Ollama 优化的“出生证明”。

操作步骤:

  1. 进入目录
cd D:\Code\LoRA\models\Elaine_Final_Model 
  1. 创建模型
    运行以下命令,让 Ollama 识别并加载该模型:
ollama create Elaine -f Modelfile 
在这里插入图片描述
  1. 运行测试
    现在,你可以关闭任何 Python 脚本,直接在系统终端呼唤 Elaine:
ollama run Elaine 
在这里插入图片描述

6.3 最终成果验证

至此,Elaine 已经成为了你电脑里的一个系统级服务。你可以通过以下方式验证她的“自由身”:

  • 多端访问:你可以使用任何支持 Ollama 协议的客户端(如 Chatbox、Page Assist 浏览器插件)连接本地 Ollama,选择 Elaine 模型进行对话。
  • 独立性:你可以将 Elaine_Final_Model 文件夹拷贝到任何安装了 Ollama 的电脑上,重复上述 create 步骤,无需再次安装 LLaMA-Factory 或复杂的 Python 依赖。
  • 过本次实验,我们成功地在消费级显卡(8G 显存)环境下,完成了一个专属垂直领域大模型的全链路开发。从数据清洗到模型部署,我们不仅构建了名为
    Elaine 的 AI 助手,更掌握了一套可复用的低成本微调方案。

在这里插入图片描述

🔗 源码与资源 (Repository)

  • 本项目的完整代码、配置文件及数据集处理脚本已同步至 GitHub,欢迎 Star 或 Fork 进行二次开发:
  • GitHub仓库Elaine-one/Qwen2.5-LoRA
  • 主要内容:包含全套微调脚本、环境变量配置等。

Read more

【AI】——SpringAI通过Ollama本地部署的Deepseek模型实现一个对话机器人(二)

【AI】——SpringAI通过Ollama本地部署的Deepseek模型实现一个对话机器人(二)

🎼个人主页:【Y小夜】 😎作者简介:一位双非学校的大三学生,编程爱好者, 专注于基础和实战分享,欢迎私信咨询! 🎆入门专栏:🎇【MySQL,Javaweb,Rust,python】 🎈热门专栏:🎊【Springboot,Redis,Springsecurity,Docker,AI】  感谢您的点赞、关注、评论、收藏、是对我最大的认可和支持!❤️ 目录 🎈Java调用Deepseek  🍕下载Deepseek模型  🍕本地测试  🍕Java调用模型 🎈构建数据库  🍕增强检索RAG  🍕向量数据库  🍕Springboot集成pgvector 🎈chatpdf 🎈function call调用自定义函数 🎈多模态能力 🎈Java调用Deepseek 本地没有安装Ollama、Docker,openwebUI,可以先学习一下这篇文章:【AI】——结合Ollama、Open WebUI和Docker本地部署可视化AI大语言模型_ollma+本地大模型+open web ui-ZEEKLOG博客

OmniSteward:LLM Agent 赋能,语音文字随心控,智能家居与电脑的超级管家

OmniSteward:LLM Agent 赋能,语音文字随心控,智能家居与电脑的超级管家

目录 * 一、前言 * 二、项目概述 * 三、功能特性 * 四、技术架构 * 五、安装与使用 * 1、系统要求 * 2、安装步骤 * 3、环境变量配置 * 4、启动方式 * 4.1 命令行模式(CLI) * 4.2 Web模式 * 六、应用场景与未来展望 * 七、结语 一、前言 在科技日新月异的今天,人工智能正以前所未有的速度改变着我们的生活方式。从智能手机的语音助手到智能家居的自动化控制,AI技术逐渐渗透到生活的各个角落,为我们带来了便捷与高效。OmniSteward正是在这样的背景下应运而生,它作为一款基于大语言模型的全能AI管家系统,致力于打破人机交互的壁垒,为用户打造一个智能、高效、便捷的生活和工作环境。无论是忙碌的上班族希望在工作中提高效率,还是追求高品质生活的家庭用户渴望轻松掌控家居设备,OmniSteward都有可能成为他们理想的智能伙伴,引领我们进入一个全新的智能生活时代。 二、项目概述 OmniSteward是一个正在积极开发中的全能管家系统,

【UAV】基于射频和深度学习的无人机检测与识别:构建大型开源无人机数据库的初步探索【附MATLAB+python代码】

【UAV】基于射频和深度学习的无人机检测与识别:构建大型开源无人机数据库的初步探索【附MATLAB+python代码】

文章来源:微信公众号:EW Frontier QQ交流群:1074124098(如满私信留言获取最新群) 注:本文为参考文章~代码为对其部分内容进行复现~ 若有侵权联系删除。 摘要 民用无人机的普及可能引发技术、安全和公共安全问题,这些问题亟待解决、规范和预防。安全机构一直在不断探索能够检测无人机的技术和智能系统。然而,相关技术的突破因缺乏无人机射频信号开源数据库而受阻,这些射频信号可通过远程传感和存储,为开发最有效的无人机检测与识别方法提供支持。本文为构建不同飞行模式下各类无人机射频信号数据库迈出了重要一步。我们系统地收集、分析并记录了不同无人机在不同飞行模式(如关闭、开启并连接、悬停、飞行和视频录制)下的原始射频信号。此外,我们利用所开发的射频数据库设计了智能算法,用于检测和识别入侵无人机。研究采用三个深度神经网络,分别实现无人机存在检测、无人机存在及类型识别,以及无人机存在、类型及飞行模式识别。通过 10 折交叉验证过程验证每个深度神经网络的性能,并使用多种指标进行评估。分类结果显示,随着类别数量的增加,性能普遍下降。平均准确率从第一个深度神经网络(2 类)的 99.7%