LLaMA-Factory正式支持百度文心全系列,实战单卡微调宝可梦图鉴

59 阅读3分钟

就在刚刚,百度飞桨官方发布,文心大模型和 LLaMA-Factory 达成生态合作。

现在可以“开箱即用”地支持 ERNIE-4.5 全系列,甚至把 28B 参数量的 ERNIE-4.5-VL 模型微调门槛降到了单张消费级显卡(约 20GB 显存)。

官方放了一个有趣的实战案例,“复刻宝可梦图鉴”。我研究了一下,在这里我把其中的核心逻辑给大家扒一扒。

官方实测环境:A100 下的 40 分钟速通

根据官方披露的数据,利用 LLaMA-Factory 的 4-bit QLoRA 能力,在 NVIDIA A100 环境下,针对前 50 个宝可梦数据集,全流程微调耗时仅需约 40 分钟。

Step 1:环境搭建与模型准备

首先强调的是环境的一致性。因为要支持 ERNIE-4.5-VL,官方使用了开发版的 Transformers 和与之匹配的 ERNIE 模型。

他们提供的初始化脚本非常简洁:

`Bash**

# 浅层克隆 LLaMA-Factory 最新版
!git clone --depth 1 https://github.com/hiyouga/LLaMA-Factory.git
%cd LLaMA-Factory

# 安装核心依赖(含 bitsandbytes 和 deepspeed)
!pip install -e ".[torch,metrics,bitsandbytes,deepspeed]"
!pip install git+https://github.com/huggingface/transformers.git

# 从 HF 下载官方指定的 ERNIE 模型版本
from huggingface_hub import snapshot_download
snapshot_download(repo_id="baidu/ERNIE-4.5-VL-28B-A3B-PT", revision="refs/pr/10")

Step 2:数据构造(ShareGPT 格式)

在数据处理上,官方演示了如何将图文数据转化为 LLaMA-Factory 支持的 ShareGPT 格式。这对于想做多模态微调的朋友很有参考价值:需要在 conversations 的 value 字段中加入 标记。

官方给出的 JSON 样本和注册配置如下:

JSON

/* 训练数据片段 */
{
"images": ["/data/images/pokemon_019.jpg"],
"conversations": [
{ "from": "human", "value": "<image>\n这个宝可梦是谁?请描述它。" },
{ "from": "gpt", "value": "妙蛙种子:这是一种小型、四足动物..." }
]
}

准备好数据后,只需在 dataset_info.json` 里完成注册即可被框架识别。

Step 3:核心配置 YAML

这是整个教程里我最喜欢的部分,配置即代码。官方通过一个 YAML 文件就完成了所有参数设置,不用写繁琐的训练循环。

注意看他们为了适配消费级显卡做的关键配置:

  • template: ernie_vl:必须指定这个专用模板。
  • quantization_bit: 4:开启 4-bit 量化,这是把 28B 模型塞进 20GB 显存的关键。

`YAML

### 官方提供的 ernie_vl_sft.yaml 核心片段
model_name_or_path: path_to_ernie_local
template: ernie_vl
finetuning_type: lora
lora_target: q_proj,v_proj
dataset: pokemon_50
quantization_bit: 4  # 显存优化核心
output_dir: saves/ernie_pokemon

Step 4 & 5:一键训练与推理验证

配置完成后,官方演示的启动命令非常简单:

llamafactory-cli train ernie_vl_sft.yaml

训练结束后,为了验证效果,官方还提供了一段基于 ChatModel` 接口的推理代码。对比效果非常明显:

  • 微调前**:模型只能泛泛识别出是“精灵宝可梦”。
  • 微调后:加载了 Adapter 权重的模型,能够精准说出角色的名字(如妙蛙种子)并描述其外观细节。

写在最后

现在的多模态微调门槛确实被 LLaMA-Factory 拉低了不少。如果你手头有显卡,可以照着这个官方教程复现一下,把自己的数据喂给 ERNIE 试试看。

感兴趣的朋友可以直接去跑一下官方提供的 Google Colab Notebook,或者去 GitHub 仓库拉一下完整代码玩一玩。

你是不是也想:摆脱朝九晚五的束缚,拥有一份 “睡后收入”?成为别人口中 “会搞钱的超级个体”,活成自己喜欢的样子?但内容创作太难、账号运营太复杂?别让 “不会” 拦住你的野心!

你要拥抱AI啊,AI是这个时代赋予我们每一个普通人翻身最好的武器,今天给大家推荐一个AI黑科技👉01agent.net?utm\_source=juejin 小白也能快速出文案、自动做物料、轻松起账号用 AI 当 “外挂”,把你的才华放大 10 倍!从副业小白到超级个体,只差一个AI 武器的距离!