PaddleMIX
PaddleMIX copied to clipboard
wuerstchen 训练支持
wuerstchen 训练支持
任务描述
任务背景
- PaddleMIX ppdiffusers新增wuerstchen训练流程。
完成步骤
- 参考代码完成对齐,若缺少相应的models部分需实现并添加到ppdiffusers
提交内容:
- 提交到目录
尝试解决该任务遇到以下问题:
- 该模型需基于Paddle引入类似’from torchvision.models import efficientnet_v2_l, efficientnet_v2_s‘,目前经调研PaddleClas中的efficientnet实现似乎缺乏torchvision中的对应版本’efficientnet_v2_l, efficientnet_v2_s‘,无法直接调研PaddleClas;
- 在当前的ppdiffusers下,未植入wuerstchen的Pipeline和Schedule实现,故完成该任务前似乎需先对wuerstchen相关的Pipeline和Schedule进行代码移植。