mindspore-lab / mindone

one for all, Optimal generator with No Exception
https://mindspore-lab.github.io/mindone/
Apache License 2.0
355 stars 69 forks source link

pangu draw里面的"DiffusionEngineMultiGraph"的作用是什么,目前好像还用不上 #316

Open ultranationalism opened 8 months ago

ultranationalism commented 8 months ago

diffusion.py下面有个DiffusionEngine的子类叫DiffusionEngineMultiGraph,但是pangu_sd_xl_base.yaml看出来并没有使用DiffusionEngineMultiGraph而是直接用了DiffusionEngine,所以我很好奇这个DiffusionEngineMultiGraph的作用

townwish4git commented 8 months ago

这是为SD XL模型的训练任务开发的一个实验性功能,由于盘古模型以SD XL为基模,因此继承了这个功能。目前我们所有任务用到的仍是DiffusionEngine,因此无需关注DiffusionEngineMultiGraph