代码实现

参考实现

url=https://github.com/huggingface/diffusers
commit_id=7f551e29ff4ad05615cb38530a8940811f9e5936

适配昇腾AI处理器的实现

url=https://gitee.com/ascend/ModelZoo-PyTorch.git
code_path=PyTorch/built-in/diffusion/diffusers0.25.0/
  • 为方便理解迁移流程,本案例给出的模型为昇腾开源仓diffusers0.25.0中的SDXL模型,环境准备与拉起训练流程等详见readme
  • 如需执行实际训练任务请跳转至MindSpeed-MM