Datawhale AI夏令营第四期 魔搭-AIGC方向 task01笔记

任务1：跑通baseline

第一步——搭建代码环境

• 使用魔搭的免费Notebook实例

• 

• 

  第二步——下载&执行baseline文件

1. git lfs install 
   git clone https://www.modelscope.cn/datasets/maochase/kolors.git 

2. 安装环境，然后重启kernel

   安装 Data-Juicer 和 DiffSynth-Studio

   Data-Juicer：数据处理和转换工具，旨在简化数据的提取、转换和加载过程

   DiffSynth-Studio：高效微调训练大模型工具

   3.依次顺序运行剩余的代码块，点击代码框左上角执行按钮，最终获得图片（大约需要20分钟）

1. 使用Data-Juicer处理数据，整理训练数据文件
2. 使用DiffSynth-Studio在基础模型上，使用前面整理好的数据文件进行训练微调
3. 加载训练微调后的模型
4. 使用微调后的模型，生成用户指定的prompt提示词的图片

 

第三步——微调结果上传

 

1.移动结果文件

创建terminal，粘贴如下命令，回车执行

mkdir /mnt/workspace/kolors/output & cd cp /mnt/workspace/kolors/models/lightning_logs/version_0/checkpoints/epoch\=0-step\=500.ckpt /mnt/workspace/kolors/output/ cp /mnt/workspace/kolors/1.jpg /mnt/workspace/kolors/output/

2.下载结果文件

双击进入output文件夹，分别下载两个文件到本地

3.创建并上传模型所需内容
 

点击魔搭链接，创建模型，中文名称建议格式：队伍名称-可图Kolors训练-xxxxxx

4.来到创空间，查看自己的模型是否发布