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模型参数正常更新