MegEngine / MegDiffusion

MegEngine implementation of Diffusion Models.
Apache License 2.0
16 stars 0 forks source link

feat: add pretrained models convert script #6

Closed Asthestarsfalll closed 2 years ago

Asthestarsfalll commented 2 years ago

没有找到更好的方法来转换,这里使用了映射字典的键名来实现转换。 实际上我认为通过观察key可以更好地了解模型内组件的层级关系,通过python tools/convert - p可以打印key的转换过程,便于理解。