Part 1 跑通baseline
-
导入库:首先,代码导入了需要用到的库,包括 data-juicer 和微调的工具 DiffSynth-Studio
Data-Juicer:数据处理和转换工具,旨在简化数据的提取、转换和加载过程
DiffSynth-Studio:高效微调训练大模型工具
-
数据集构建:下载数据集kolors,处理数据集
-
模型微调:模型微调训练,以及加载训练后的模型
-
图片生成:调用训练好的模型生成图片
导入库:首先,代码导入了需要用到的库,包括 data-juicer 和微调的工具 DiffSynth-Studio
Data-Juicer:数据处理和转换工具,旨在简化数据的提取、转换和加载过程
DiffSynth-Studio:高效微调训练大模型工具
数据集构建:下载数据集kolors,处理数据集
模型微调:模型微调训练,以及加载训练后的模型
图片生成:调用训练好的模型生成图片