LLaMA-Factory微调和部署指南

部署运行你感兴趣的模型镜像

1.conda创建自己虚拟环境并且激活

conda create --name myenv python=3.10
conda activate myenv

2.安装LLaMA-Factor

git clone --depth 1 https://github.com/hiyouga/LLaMA-Factory.git
cd LLaMA-Factory
pip install -e .

3.下载训练的大模型 (以通义千问2.5-1.5B-Instruct为例,SDK进行下载)

先下载modelscope包
pip install modelscope

#模型下载
from modelscope import snapshot_download
model_dir = snapshot_download('Qwen/Qwen2.5-1.5B-Instruct',cache_dir=r'/root/autodl-tmp/demo/model')

 4.加载自定义数据集放到/root/autodl-tmp/demo/LLaMA-Factory/data下(下面是用vscode连接)

下面修改dataset_info.json(防止WebUi进去train的Dataset选不到)

5.在LLaMA-Factory下启动WebUI(使用VSCode有自动转发功能)

llamafactory-cli webui

6.进入http://localhost:7860/中进行配置,选中自己的模型和数据集

7.点击下面开始训练,报错,那么根据提示进行安装

pip install bitsandbytes==0.37.0

8.点击开始训练,又报错,那么重新安装bitsandbytes

pip uninstall bitsandbytes -y
pip install bitsandbytes  # 尝试默认安装

9.点击开始进行训练

10.测试趋于平缓进行验证

11.验证通过进行合并打包

您可能感兴趣的与本文相关的镜像

Llama Factory

Llama Factory

模型微调
LLama-Factory

LLaMA Factory 是一个简单易用且高效的大型语言模型(Large Language Model)训练与微调平台。通过 LLaMA Factory,可以在无需编写任何代码的前提下,在本地完成上百种预训练模型的微调

评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值