零基础入门大模型微调攻略！一篇文章掌握核心技术！

玩转AI大模型

于 2024-09-05 11:05:35 发布

阅读量743

点赞数 17

本文链接：https://blog.csdn.net/2401_85375298/article/details/141925465

版权

今天分享一篇技术文章，你可能听说过很多大模型的知识，但却从未亲自使用或微调过大模型。

今天这篇文章，就手把手带你从零微调一个大模型。

大模型微调本身是一件非常复杂且技术难度很高的任务，因此本篇文章仅从零开始，手把手带你走一遍微调大模型的过程，并不会涉及过多技术细节。

希望通过本文，你可以了解微调大模型的流程。

微调大模型需要非常高的电脑配置，比如GPU环境，相当于你在已经预训练好的基础上再对大模型进行一次小的训练。

但是不用担心，本篇文章会使用阿里魔塔社区提供的集成环境来进行，无需使用你自己的电脑配置环境。

你只需要有浏览器就可以完成。

本次微调的大模型是零一万物的 Yi 开源大语言模型，当然微调其他大模型的过程和原理也有差不多。

这里说明一下，阿里魔塔社区对于新用户提供了几十小时的免费GPU资源进行使用，正好可以来薅一波羊毛，学习一下大模型的微调。

话不多说，直接开始。

1. 账号和环境准备

首先你需要注册和登录魔搭的账号：https://modelscope.cn/home

注册完成后，登录这个模型网址：

https://www.modelscope.cn/models/01ai/Yi-1.5-6B.

然后按照下面的箭头操作。

选择完方式二：GPU环境后，点击“启动”。

启动大概需要2分钟，等GPU环境启动好以后点击"查看NoteBook"进入。

魔塔社区内置了JupyterLab的功能，你进入之后，可以找到 Notebook 标签，新建一个Notebook(当然你在terminal 里执行也没问题)。

如下箭头所示，点击即可创建一个新的 Notebook 页面。

增添一个代码块，并且执行以下命令（点击左侧的运行按钮运行该代码块，下同，这一步是安装依赖库）。

!pip3 install --upgrade pip``!pip3 install bitsandbytes>=0.39.0

拉取 LLaMA-Factory，过程大约需要几分钟

!git clone --depth 1 https://github.com/hiyouga/LLaMA-Factory.git

接下来需要去 Launcher > Terminal 执行（按照图片剪头指示操作）。

安装依赖的软件，这步需要的时间比较长。

# ⚠️下面两行命令在刚启动的Terminal中执行⚠️``cd LLaMA-Factory``pip3 install -e ".[torch,metrics]"

等以上所有步骤完成后，再进行下面的操作。

2. 下载模型

零一万物的 Yi 开源大语言模型的权重可以在HuggingFace和ModelScope上找到，这里我选择从ModelScope上下载。

零一万物的所有开源模型链接在这里：

https://www.modelscope.cn/organization/01ai/

模型下载需要一定的时间，这里选择了最小的Yi-1.5-6B-chat模型进行实验。

模型的说明在这里：

https://www.modelscope.cn/models/01ai/Yi-1.5-6B-Chat/summary

Yi-1.5-6B-chat模型大小大约12G，下载大约需要10分钟（取决于网速）。

接下来，你通过下面的命令就可以在 notebook 里执行下载（在 terminal也一样，如果需要在terminal执行需要去掉前面的!）。

!git clone https://www.modelscope.cn/01ai/Yi-1.5-6B-Chat.git

这一步，耐心等待下载完成即可。

3. 微调Yi模型实战

等以上所有步骤完成后，准备工作就做好了，现在可以开始准备微调了。

⚠️注意：虽然本篇文章仅仅是简单的过一遍微调的流程，但是不要低估他的难度。微调跑起来很容易，但是跑出很‍好的结果非常的难。

开源社区有许多非常优秀的专门用于微调代码库具体的你可以参考这里：

https://github.com/01-ai/Yi-1.5?tab=readme-ov-file#fine-tuning

站在巨人的肩膀上开始这次实战，这里选择llama_factory。

LLaMA Factory是一款开源低代码大模型微调框架，集成了业界广泛使用的微调技术。llama_factory 的介绍可以在这里查看：

https://github.com/hiyouga/LLaMA-Factory

4. 开始微调

a. 创建微调训练相关的配置文件

在左侧的文件列表，Llama-Factory的文件夹里，打开examples\train_qlora（注意不是 train_lora）下提供的llama3_lora_sft_awq.yaml，复制一份并重命名为yi_lora_sft_bitsandbytes.yaml。

这个文件里面写着和微调相关的关键参数。

打开这个文件，将第一行model_name_or_path更改为你下载模型的位置。

### model``model_name_or_path: <你下载的模型位置，不要带括号，比如我写了../Yi-1.5-6B-Chat>

同样修改其他行的内容，下面是我的修改，你可以逐行对比一下，有不一致或缺少的就添加一下。

### model``model_name_or_path: ../Yi-1.5-6B-Chat``quantization_bit: 4``   ``### method``stage: sft``do_train: true``finetuning_type: lora``lora_target: all``   ``### dataset``dataset: identity``template: yi``cutoff_len: 1024``max_samples: 1000``overwrite_cache: true``preprocessing_num_workers: 16``   ``### output``output_dir: saves/yi-6b/lora/sft``logging_steps: 10``save_steps: 500``plot_loss: true``overwrite_output_dir: true``   ``### train``per_device_train_batch_size: 1``gradient_accumulation_steps: 8``learning_rate: 1.0e-4``num_train_epochs: 3.0``lr_scheduler_type: cosine``warmup_ratio: 0.1``bf16: true``# ddp_timeout: 180000000``   ``### eval``val_size: 0.1``per_device_eval_batch_size: 1``eval_strategy: steps``eval_steps: 500

从上面的配置文件中可以看到，本次微调的数据集是 identity。

那这个文件里面写着什么呢？

你可以打开这个文件看一下：https://github.com/hiyouga/LLaMA-Factory/blob/main/data/identity.json。

微调数据集是“自我认知”，也就是说当你问模型“你好你是谁”的时候，模型会告诉你我叫name由author开发。

如果你把数据集更改成你自己的名字，那你就可以微调一个属于你自己的大模型。

这一步，你可以将 identity.json 中的 {{name}} 字段替换为你的名字来微调一个属于自己的大模型。

保存刚才对于 yi_lora_sft_bitsandbytes.yaml 文件的更改，回到终端terminal。

在 LLaMA-Factory 目录下，输入以下命令启动微调脚本(大概需要10分钟)

llamafactory-cli train examples/train_qlora/yi_lora_sft_bitsandbytes.yaml

看到进度条就是开始微调了。

运行过程大概需要10分钟，当你看到下面这个界面的时候，微调过程就结束了。

5. 推理测试

微调后的模型有什么不同的地方呢？

这里加载微调后模型进行推理，测试微调前后变化。

参考Llama-Factory文件夹中，examples\inference下提供的llama3_lora_sft.yaml，复制一份，并重命名为 yi_lora_sft.yaml

将内容更改为，并且保存（一定记得保存）****。

model_name_or_path: <和之前一样，你下载的模型位置，比如我写了../Yi-1.5-6B-Chat>``adapter_name_or_path: saves/yi-6b/lora/sft``template: yi``finetuning_type: lora

回到刚刚结束微调的终端Terminal，运行下面的推理命令(同样在Llama-Factory目录下运行）。

llamafactory-cli chat examples/inference/yi_lora_sft.yaml

稍微等待一下模型加载，然后就可以聊天了。

可以看到模型的自我身份认知被成功的更改了。

自我身份认知更改成为数据集规定的样子了，同时也保持了通用对话能力。

那么，和没有经过微调之前的模型对比有什么差别呢？

重复上面的步骤，将llama3.yaml复制并重命名为yi.yaml，将内容更改为以下的内容，并保存（一定记得保存）。

model_name_or_path: ../Yi-1.5-6B-Chat``template: chatml

回到终端Terminal，运行下面的推理命令：

llamafactory-cli chat examples/inference/yi.yaml

可以提问和刚才同样的问题，看到模型的原始回答。

基于本实验，你就完成了一个简单的微调，完整的走了一遍模型的微调过程，是不是还挺简单的？

如何学习大模型 AI ？

由于新岗位的生产效率，要优于被取代岗位的生产效率，所以实际上整个社会的生产效率是提升的。

但是具体到个人，只能说是：

“最先掌握AI的人，将会比较晚掌握AI的人有竞争优势”。

这句话，放在计算机、互联网、移动互联网的开局时期，都是一样的道理。

我在一线互联网企业工作十余年里，指导过不少同行后辈。帮助很多人得到了学习和成长。

我意识到有很多经验和知识值得分享给大家，也可以通过我们的能力和经验解答大家在人工智能学习中的很多困惑，所以在工作繁忙的情况下还是坚持各种整理和分享。但苦于知识传播途径有限，很多互联网行业朋友无法获得正确的资料得到学习提升，故此将并将重要的AI大模型资料包括AI大模型入门学习思维导图、精品AI大模型学习书籍手册、视频教程、实战学习等录播视频免费分享出来。

😝有需要的小伙伴，可以Vx扫描下方二维码免费领取🆓