Qwen-Image2.1Lora训练AITOOLKIT
使用 AI Toolkit 训练千问 Image 2.1 的编辑 LoRA,本教程示例为二次元转真人风格转换,训练效果明显优于千问原生模型直接处理的效果。
| 精度 | 推荐显卡 | 说明 |
|---|---|---|
| BF16(原生精度) | 4090 及以上 | 效果更好,但显存占用更高 |
| INT8 | 3090 即可 | 显存占用小,但效果比 BF16 差不少,速度也慢约 3 倍 |
⚠️ 两者二选一:追求效果用 BF16(4090+),显卡有限用 INT8(3090 可用)
# 1. 部署 GPU 实例(4090 或 3090,按精度选择)
# 2. 打开 Jupyter Lab,复制启动指令粘贴运行
# 3. 打开 AI Toolkit
编辑 LoRA 需要一一对应的两组图片:
⚠️ Control 和 Target 的文件命名必须一一对应。直接在 Dataset 页面内上传比新建文件夹再上传更快。
不需要打标,Dataset 类型分别选 Control 和 Target,默认提示词留空,训练时用触发词即可,不需要额外打标。
| 参数 | 推荐值 | 说明 |
|---|---|---|
| 精度 | BF16(4090)/ INT8(3090) | 按显卡选择,显存不够就用 INT8 |
| 触发词 | 自定义(如 photo realistic live action) | 描述目标风格即可 |
| 优化器 | Automagic v3(推荐) | 显存紧张可用 Adam 8bit;训练角色类可用 Prodigy |
| 分辨率 | 1280(显存够用时) | 报显存不足就降到 512 |
| Layer Offloading | 报显存不足就开启,并调高(如从 30 起步逐步上调) | 采样阶段最吃显存,容易在这一步爆掉 |