4060显卡+LLaMA-Factory微调LLM——(四)Web UI

本地配置

系统:win10
**硬件:**i5-12490f+RTX4060

前置准备

目前本地微调大模型最常用的就是LLaMA-Factory,既可以用指令微调,又有Web UI,方便的很
github:https://github.com/hiyouga/LLaMA-Factory
首先将克隆到本地

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

打开conda prompt,为什么不是cmd,因为我的pytorch等都是安装再conda环境下的,想利用cuda跑大模型,所以打开的是conda,而不是cmd
image.png
创建一个conda环境,这里说下,conda默认是base环境,可以新建环境,也可以就在base环境上面跑

conda create -n llm python=3.0

安装pytorch,已经安装过的可以忽略

conda install pytorch torchvision torchaudio pytorch-cuda=12.1 -c pytorch -c nvidia


进入最开始clone LLama-factory的文件夹下,运行下面两条

pip install -e .[torch,metrics]
pip install https://github.com/jllllll/bitsandbytes-windows-webui/releases/download/wheels/bitsandbytes-0.41.2.post2-py3-none-win_amd64.whl




然后逐一运行下面的指令,你就可以看到打开webui的链接了

Set CUDA_VISIBLE_DEVICES=0
Set GRADIO_SHARE=1
llamafactory-cli webui


本地打开链接后,出现下面的界面

这里可以选择模型的名称进行下载,然后再微调
具体的微调参数不在这里做过多介绍,官网都有

有几个点需要注意:

  1. 下载大模型的时候,如果报错,可能是你网络不行,没办法直连到huggingface上面进行下载,那你就只能去modelscope下载模型了
  2. 硬件配置,参考官网的硬件要求来,不要太过勉强,小心显卡冒烟

image.png

  1. 不要一上来就训练很多轮,epochs设置建议由小到大去进行试探,如果你的显存已经满了,而epochs又很大的话,那么训练的速度会非常非常慢
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值