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

cnblogs 2024-10-16 16:13:00 阅读 55

环境搭建

系统环境

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

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

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

设置自定义Path文件夹

创建 <code>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_MODELSE:\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

模型可以使用了~



声明

本文内容仅代表作者观点,或转载于其他网站,本站不以此文作为商业用途
如有涉及侵权,请联系本站进行删除
转载本站原创文章,请注明来源及作者。