先是跑通baseline,建好模型后,根据所得到的模型以及自己加上去的一些特征词之类的来得到图像
baseline代码的大致结构
-
导入库:首先,代码导入了需要用到的库,包括 data-juicer 和微调的工具 DiffSynth-Studio
-
数据集构建:下载数据集kolors,处理数据集
-
模型微调:模型微调训练,以及加载训练后的模型
-
图片生成:调用训练好的模型生成图片
夏令营链接:Datawhale (linklearner.com)
先是跑通baseline,建好模型后,根据所得到的模型以及自己加上去的一些特征词之类的来得到图像
baseline代码的大致结构
导入库:首先,代码导入了需要用到的库,包括 data-juicer 和微调的工具 DiffSynth-Studio
数据集构建:下载数据集kolors,处理数据集
模型微调:模型微调训练,以及加载训练后的模型
图片生成:调用训练好的模型生成图片
夏令营链接:Datawhale (linklearner.com)