作为第一次接触AI绘画得小白,先试着跑一下baseline。
第一步:安装
安装 Data-Juicer 和 DiffSynth-Studio
第二步:下载数据集
保存数据集中的图片及元数据
第三步:数据处理
使用 data-juicer 处理数据
保存处理好的数据
第四步:训练模型
下载模型
开始训练
提示:
- 在训练命令中填入
--modelscope_model_id xxxxx
以及--modelscope_access_token xxxxx
后,训练程序会在结束时自动上传模型到 ModelScope - 部分参数可根据实际需求调整,例如
lora_rank
可以控制 LoRA 模型的参数量 - 生成图像
跑完baseline得到了结果
可以得到精美得兔子得图片,经过观察可以发现,图片之间得关联性不大,兔子前后长相不一致,因此将来可以对网络结果进行更改加入时序信息,初次之外,还要优化数据集。