Author

Hobart(一个图像生成领域的代码学习者)

项目背景

根据Step1X-Edit(https://github.com/stepfun-ai/Step1X-Edit)推理代码，对训练代码进行复现

代码执行

下载Step1X-Edit(https://github.com/stepfun-ai/Step1X-Edit)

将Step1X-Edit_train.py放入Step1X-Edit项目下

python Step1X-Edit_train.py

功能描述

简单实现Step1X-Edit的训练

数据只demo读取了一下，私人的数据集读取格式需要修改

代码已经验证dit模型参数正常更新