PaddlePaddle/PaddleMIX

wuerstchen 训练支持

chenjjcccc opened this issue · 1 comments

wuerstchen 训练支持

任务描述

任务背景

  • PaddleMIX ppdiffusers新增wuerstchen训练流程。

完成步骤

  1. 参考代码完成对齐,若缺少相应的models部分需实现并添加到ppdiffusers

提交内容:

  1. 提交到目录

尝试解决该任务遇到以下问题:

  • 该模型需基于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进行代码移植。