3分钟微调大模型!快速微调Llama3.1-8B

AI浪潮席卷全球并发展至今已有近2年的时间了,大模型技术作为AI发展的底座和基石,更是作为AI从业者必须掌握的技能。但是作为非技术人员,相信大家也有一颗想要训练或微调一个大模型的心,但是苦于技术门槛太高,无从下手。今天教大家一个非常快速的方法,5分钟就可以让你快速上手去微调一个大模型。还不赶紧去试试!

什么是模型微调?

大模型微调(Fine-tuning)是指在已经预训练好的大型深度学习模型基础上,使用新的、特定任务相关的数据集对模型进行进一步训练的过程。这里我们用的微调方式选择的是Lora。

首先,我们是要基于一个开源的大模型去微调一个属于我们自己模型,所以我们要先找一个开源的大模型。而开源大模型社区,现在最火的一定是Hugging Face。但是苦于需要魔法上网,所以我们用国内的知名模型库魔搭社区。

网址:

https://www.modelscope.cn/models

比如我们接下来要演示的是近期刚发布的Meta Llama3.1-8B的一个大模型。

模型地址:

https://www.modelscope.cn/models/LLM-Research/Meta-Llama-3.1-8B-Instruct

如何快速微调一个模型呢,下载提供好的训练模型代码后,只需要准备训练数据集和修改你要训练的模型即可,是不是特别快,特别方便。

1、下载训练模型的代码

我已经整理好了,在公众号回复【Ft-models】即可获取。解压后目录如下:

(解压后)

train.py:要训练的脚本。

test.py:训练后用于测试的脚本。

chat.py:基于streamlit的简单web问答交互页面。

requirements.txt:运行时需要的python库文件。

dataset:存放要训练的数据。

models:存放模型(默认为空,启动train.py进行训练时会下载模型到该目录)

2、准备数据集

需要准备要训练的数据集,json格式的,为了训练快速,我只象征性的加了3条。后续大家根据自己需求添加即可。

(准备json格式数据集)

3、修改训练脚本(train.py) & 训练

修改训练脚本中的模型,这里选用的Llama-3.1-8B,想微调其他的模型,在魔搭社区查找完修改即可。

(修改要训练的模型名称)

修改训练的设备,因为我是mac,所以写死了用mps,大家如果用gpu训练的话,改成cuda(在此之前电脑别忘了安装驱动,但先要确认NVIDIA版本是否支持CUDA),否则值设成用cpu进行训练。

总结一下,这里的可选值有三个:mps、cuda、cpu。

(修改要训练的硬件用什么)

到这里就可以执行【train.py】进行训练了,训练的过程第一次会很慢,因为需要在线下载模型和训练代码依赖的python库。后续再次进行训练就很快了,后续的训练速度取决于你训练数据集的大小和电脑硬件的配置,用GPU肯定比CPU快多了。

训练前需要安装python环境,然后执行下面命令进行训练:

python3 train.py

(模型训练)

注意:因为我准备的训练数据只有三条,很少,只用于测试,为了训练结果更精确,所以训练的总轮数(num_train_epochs这个参数)我设定为了30,大家如果后续使用时,这个参数根据需要可以调整。

4、训练后测试

训练后的模型如何加载和使用呢?这里提供了两种方式,第一种是写了一个【test.py】的python脚本去调用。第二种是通过【chat.py】去调用,里面封装了用streamlit写的web页面。

第一种方式【test.py】:

和上面一样,打开【test.py】,需要修改要加载的模型和运行的硬件(可选值有mps、cuda、cpu)不多说了。

(修改要加载的模型名称)

(修改要推理的硬件)

运行下面代码进行第一种方式运行【test.py】进行测试:

python3 test.py

为了能看出来我们微调后模型是否有有效果,所以在训练前,我截了一个训练前的效果图。

微调训练前回答效果:

(微调前回答效果)

当我问:【什么是大模型?】

给我的回答是:【大模型(Large Model)指的是在机器学习和深度学习领域中,训练和应用的模型规模非常庞大和复杂的模型。这些模型通常由大量参数组成,能处理和学习复杂的数据模式和关系】

完全基于之前大模型自身的能力进行回答,解释了什么是大模型。

微调训练后回答效果:

(微调后回答效果)

当我问:【什么是大模型?】

给我的回答是:【LLM(Large Language Model),一般指万万级参数以上的模型,但是这个标准一直在升级,目前万亿参数以上的模型也有了。一般万万级参数以上的模型万亿参数以下的模型也有,但是这个标准一直在升级升级。】

这次回答是基于我们微调后进行回答的**,因为在数据集中我们进行了定义。所以微调是成功的**。(代码中现在把输出的max_new_tokens设定为定值100,所以会输出的结果有重复的情况)

第二种方式【chat.py】:

通过【chat.py】去调用,里面封装了用streamlit写的web页面。

运行下面代码进行第二种方式运行【chat.py】进行测试:

streamlit run chat.py

运行后会打开Web的Chatbot页面,可在页面进行提问和交互。

(基于streamlit的简版Chatbot)

如何学习AI大模型 ?

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

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

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

我意识到有很多经验和知识值得分享给大家,故此将并将重要的AI大模型资料包括AI大模型入门学习思维导图、精品AI大模型学习书籍手册、视频教程、实战学习等录播视频免费分享出来。【保证100%免费】🆓

对于0基础小白入门:

如果你是零基础小白,想快速入门大模型是可以考虑的。

一方面是学习时间相对较短,学习内容更全面更集中。
二方面是可以根据这些资料规划好学习计划和方向。

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

👉1.大模型入门学习思维导图👈

要学习一门新的技术,作为新手一定要先学习成长路线图,方向不对,努力白费。

对于从来没有接触过AI大模型的同学,我们帮你准备了详细的学习成长路线图&学习规划。可以说是最科学最系统的学习路线,大家跟着这个大的方向学习准没问题。(全套教程文末领取哈)
在这里插入图片描述

👉2.AGI大模型配套视频👈

很多朋友都不喜欢晦涩的文字,我也为大家准备了视频教程,每个章节都是当前板块的精华浓缩。

在这里插入图片描述
在这里插入图片描述

👉3.大模型实际应用报告合集👈

这套包含640份报告的合集,涵盖了AI大模型的理论研究、技术实现、行业应用等多个方面。无论您是科研人员、工程师,还是对AI大模型感兴趣的爱好者,这套报告合集都将为您提供宝贵的信息和启示。(全套教程文末领取哈)

在这里插入图片描述

👉4.大模型落地应用案例PPT👈

光学理论是没用的,要学会跟着一起做,要动手实操,才能将自己的所学运用到实际当中去,这时候可以搞点实战案例来学习。(全套教程文末领取哈)

在这里插入图片描述

👉5.大模型经典学习电子书👈

随着人工智能技术的飞速发展,AI大模型已经成为了当今科技领域的一大热点。这些大型预训练模型,如GPT-3、BERT、XLNet等,以其强大的语言理解和生成能力,正在改变我们对人工智能的认识。 那以下这些PDF籍就是非常不错的学习资源。(全套教程文末领取哈)
img

在这里插入图片描述

👉6.大模型面试题&答案👈

截至目前大模型已经超过200个,在大模型纵横的时代,不仅大模型技术越来越卷,就连大模型相关的岗位和面试也开始越来越卷了。为了让大家更容易上车大模型算法赛道,我总结了大模型常考的面试题。(全套教程文末领取哈)

在这里插入图片描述
👉学会后的收获:👈
基于大模型全栈工程实现(前端、后端、产品经理、设计、数据分析等),通过这门课可获得不同能力;

能够利用大模型解决相关实际项目需求: 大数据时代,越来越多的企业和机构需要处理海量数据,利用大模型技术可以更好地处理这些数据,提高数据分析和决策的准确性。因此,掌握大模型应用开发技能,可以让程序员更好地应对实际项目需求;

基于大模型和企业数据AI应用开发,实现大模型理论、掌握GPU算力、硬件、LangChain开发框架和项目实战技能, 学会Fine-tuning垂直训练大模型(数据准备、数据蒸馏、大模型部署)一站式掌握;

能够完成时下热门大模型垂直领域模型训练能力,提高程序员的编码能力: 大模型应用开发需要掌握机器学习算法、深度学习

这份完整版的 AI 大模型学习资料已经上传优快云,朋友们如果需要可以微信扫描下方优快云官方认证二维码免费领取【保证100%免费

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

### 使用 LLaMA-Factory 对 LLaMA3.1 模型进行微调 为了使用 LLaMA-Factory 对 LLaMA3.1 进行微调,需遵循一系列配置和命令来设置环境并启动训练过程。 #### 设置 GPU 环境 确保已安装适合的 CUDA 版本以及 PyTorch 的 GPU 支持版本。这可以通过访问 PyTorch 官网获取相应指令完成安装[^3]: ```bash conda install pytorch torchvision torchaudio pytorch-cuda=12.1 -c pytorch -c nvidia ``` #### 验证安装 在确认所有依赖项正确无误之后,在 LLaMA-Factory 路径下调用以下命令以检验安装情况: ```bash llamafactory-cli version llamafactory-cli train -h ``` #### 准备数据集与模板文件 对于特定应用领域或任务的数据准备至关重要。假设已经准备好用于微调的数据集,并将其放置于 `data` 文件夹内。另外,还需编辑身份信息以便更好地适配自定义需求[^5]: ```python import json %cd /content/LLaMA-Factory/ NAME = "Gavin大咖打造的Llama3人工智能助手" AUTHOR = "LLaMA Factory" with open("data/identity.json", "r", encoding="utf-8") as f: dataset = json.load(f) for sample in dataset: sample["output"] = sample["output"].replace("NAME", NAME).replace("AUTHOR", AUTHOR) with open("data/identity.json", "w", encoding="utf-8") as f: json.dump(dataset, f, indent=2, ensure_ascii=False) ``` #### 启动 Web UI 或者直接运行微调脚本 有两种方式来进行实际的微调操作:通过图形界面 (WebUI) 或者命令行工具执行。这里提供两种方法的选择依据个人偏好而定。 ##### 方法一:利用 WebUI 方便调试 开启 WebUI 前先设定好使用的模型仓库为 ModelScope[^2]: ```bash export USE_MODELSCOPE_HUB=1 && llamafactory-cli webui ``` ##### 方法二:直接调用 CLI 工具快速上手 此法适用于熟悉命令行操作的用户群体,只需指定必要的参数即可开始训练进程[^1]: ```bash CUDA_VISIBLE_DEVICES=1 \ llamafactory-cli webchat \ --model_name_or_path [your path]/llm/Meta-Llama-3.1-8B-Instruct/ \ --adapter_name_or_path [your path]/llm/LLaMA-Factory/saves/Llama-3.1-8B/lora/sft-3/ \ --template llama3 \ --finetuning_type lora ``` 请注意替换 `[your path]` 为具体的本地存储位置。 #### 参数调整建议 由于不同应用场景下的最优超参可能有所差异,因此推荐根据具体情况进行适当调节。虽然不存在绝对标准的最佳实践指南,但可以根据以往经验总结出一些较为合理的区间范围作为参考起点[^4]。
评论 3
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值