使用LLaMA-Factory训练LLM大模型并用ollama调用

环境搭建

系统环境

需要Nvidia显卡，至少8G显存，且专用显存与共享显存之和大于20G

建议将非安装版的环境文件都放到非系统盘，方便重装或移植

以Windows11为例，非安装环境文件都放在 E 盘下

设置自定义Path文件夹

创建 E:\mypath 文件夹，将其添加进用户环境变量Path中，之后会用

CMake

下载 CMake 的 Windows x64 ZIP 对应文件：Download CMake

解压到E:\environment\cmake

将E:\environment\cmake\bin添加到用户环境变量Path中

C++编译

下载 Community 版 Visual Studio：Visual Studio 2022 IDE

运行后选择桌面c++板块内容安装，若中途取消安装了，可以在开始菜单下栏的新增项目（或推荐项目）中找到该安装程序

这里只需要c++的编译环境，把除了C++板块以外的安装项目全部取消勾选（该板块自动勾选系统SDK等其他相关组件，不要取消勾选）

python环境

python安装

下载新版Anaconda安装程序：Download Anaconda

安装过程中将添加到环境变量等选项全部打勾

用uv管理pip包

将uv所有文件装至 E:\uv 下

安装

添加 python 3.11 独立文件

uv python install 3.11
uv python list # 查看是否安装成功

Git环境

安装Git：Git - 安装 Git (git-scm.com)

Nvidia CUDA 工具包

安装12.1版：CUDA Toolkit 12.1 Update 1 Downloads | NVIDIA Developer

注意运行安装程序后在安装选单界面只安装CUDA驱动和工具包，不选其他显卡驱动和Nsignt

安装完后重启电脑

输入指令查看信息

nvidia-smi # 看当前驱动最大可支持的CUDA版本
nvcc -V # 看当前安装的CUDA运行时版本

部署训练框架

部署LLaMA-Factory

拉取源码

在 E:\AI 文件夹下拉取 LLaMA-Factory.git，注意挂代理加速

git config --global http.https://github.com.proxy socks5://127.0.0.1:1080 # 对github设置socks5代理
git clone --depth 1 https://github.com/hiyouga/LLaMA-Factory.git
cd LLaMA-Factory
uv venv -p 3.11
uv pip install -e ".[torch,metrics]"

安装flash-attention训练加速

在当前目录（LLaMA-Factory）下继续操作

uv pip install https://github.com/bdashore3/flash-attention/releases/download/v2.6.3/flash_attn-2.6.3+cu123torch2.3.1cxx11abiFALSE-cp311-cp311-win_amd64.whl

如果下载慢可以在url前加https://ghproxy.cn/用国内节点下载

安装hqq用于量化

在当前目录（LLaMA-Factory）下继续操作

uv pip install hqq

[!tip] 若提示UTF8相关报错解决

开启使用UTF-8提供全球语言支持

打开控制面板，更改系统区域设置，勾选Beta 版:使用Unicode UTF-8 提供全球语言支持(U)，这个在安装hqq包时用到

重启电脑，安装完后可将该项取消勾选

安装与当前cuda适配的pytorch

在当前目录（LLaMA-Factory）下继续操作

全局代理加速的情况：

参考 Start Locally | PyTorch

以下直接贴出代码

uv pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121

没有全局代理加速的情况：

先用代理加速去下载torch-2.3.1+cu121-cp311-cp311-win_amd64.whl

uv pip install .\torch-2.3.1+cu121-cp311-cp311-win_amd64.whl
uv pip install torchvision==0.18.1+cu121 torchaudio==2.3.1+cu121 -f https://download.pytorch.org/whl/torch_stable.html

配置之前保存的参数

在当前目录（LLaMA-Factory）下继续操作

修改 .\data\dataset_info.json 文件，在开头花括号后加入

  "alpaca_dataset": {
    "file_name": "alpaca_dataset.json"
  },

然后将自定数据集名称命名为 alpaca_dataset.json 放入 .\data 下

将之前保存的.yaml参数配置文件放入 .\config 下

添加环境变量

在 E:\mypath 文件夹内新建 llamafactory-cli.bat 文件，文本编辑器打开输入以下内容并保存

@echo off
cd /d E:\AI\LLaMA-Factory
.\.venv\Scripts\llamafactory-cli.exe %*

新开一个终端，输入以下指令开启web界面

llamafactory-cli webui

下载基础模型

使用带 -Chat 或 -Instruct 标签的模型作为基础模型

以 Qwen2.5-7B-Instruct 为例

到魔搭社区点下载模型：通义千问2.5-7B-Instruct · 模型库 (modelscope.cn)

按说明下载模型文件到 E:\AI\models 中（注意其中大文件下载的进度条不是实时显示，每隔一段进度才显示一次，并不是卡住不动）

模型微调

在 LLaMA-Factory 的Web界面中操作

语言选 zh

滑到最下面找到配置路径

下拉菜单选择之前保存的配置

点击载入训练参数按钮

滑到最上面

模型名称 Qwen2.5-7B-Instruct

填写本地模型路径：E:\AI\models\Qwen2.5-7B-Instruct

展开高级设置

量化等级设置4，量化方法选hqq，加速方式选flashattn2

数据集可加载多个

可选调中间位置训练轮数和最大样本数，最大样本数即要训练用的最大样本数量，点击预览数据集按钮可查看当前数据集数量，如果最大样本数小于当前数据集数量，则按最大样本数作为实际训练样本数，多出的样本不用于训练

计算类型选fp16

下面输出目录按需选填

点击开始进行训练

训练结束后查看损失图判断训练效果，损失值越小且稳定即效果好

在终端 Ctrl+C 退出Web服务

微调后的模型使用

以ollama调用为例

部署ollama

设置用户环境变量：

OLLAMA_ORIGINS			*
OLLAMA_MODELS			E:\AI\ollama_models

下载并安装：Ollama

安装后重启电脑

拉取模型

ollama pull qwen2.5:7b

微调模型格式转换

部署llama.cpp

在 E:\AI 目录下

git clone https://github.com/ggerganov/llama.cpp.git
cd llama.cpp
uv venv -p 3.11
uv pip install -r .\requirements\requirements-convert_lora_to_gguf.txt --index-strategy unsafe-best-match

模型检查点转换为gguf

在 E:\AI\llama.cpp 目录下

.venv\Scripts\activate.ps1
# python ./convert_lora_to_gguf.py --base <basemodel> <loratrain>
python ./convert_lora_to_gguf.py --base "E:\AI\models\Qwen2.5-7B-Instruct" "E:\AI\LLaMA-Factory\saves\Qwen2.5-7B-Instruct\lora\train_xxx"

转换后的gguf文件会导出到<loratrain>的路径下

部署微调模型

从ollama创建新模型

ollama 的 modelfile 模板：

# set the base model
FROM llama3:8b

# set custom parameter values
PARAMETER temperature 1
PARAMETER num_keep 24
PARAMETER stop <|start_header_id|>
PARAMETER stop <|end_header_id|>
PARAMETER stop <|eot_id|>
PARAMETER stop <|reserved_special_token

# set the model template
TEMPLATE """
{{ if .System }}<|start_header_id|>system<|end_header_id|>
{{ .System }}<|eot_id|>{{ end }}{{ if .Prompt }}<|start_header_id|>user<|end_header_id|>
{{ .Prompt }}<|eot_id|>{{ end }}<|start_header_id|>assistant<|end_header_id|>
{{ .Response }}<|eot_id|>
"""

# set the system message
SYSTEM You are llama3 from Meta, customized and hosted @ HY's Blog (https://blog.yanghong.dev).
# set Chinese lora support
ADAPTER /root/.ollama/models/lora/ggml-adapter-model.bin

将转换后的gguf文件剪切到 E:\models\mymodel 下，命名为 mymodel.gguf

在该目录下创建 mymodel.modelfile，输入以下内容并保存

FROM qwen2.5:7b

SYSTEM """
（可选设置SYSTEM）
"""

ADAPTER .\mymodel.gguf

在ollama创建该模型

ollama create mymodel -f E:\models\mymodel\mymodel.modelfile

模型可以使用了~