NCSNPP checkpoint conversion script

[karimul]

I tried to convert the cifar10 ncsnpp original checkpoint here to the diffusers model using convert_ncsnpp_original_checkpoint_to_diffusers.py but found an error:

Traceback (most recent call last): File "convert_ncsnpp_original_checkpoint_to_diffusers.py", line 171, in <module> converted_checkpoint = convert_ncsnpp_checkpoint( File "convert_ncsnpp_original_checkpoint_to_diffusers.py", line 30, in convert_ncsnpp_checkpoint new_model_architecture.time_proj.W.data = checkpoint["all_modules.0.W"].data KeyError: 'all_modules.0.W

does it not supported yet?

[patrickvonplaten]

Hey @karimul,

Hmm I think it should work, but there might very well be a bug in the conversion script. The NCSNPP checkpoints have extremely low usage at the moment so I'm not sure I'll be able to commit time into fixing this bug here. I'd be happy to review a PR though if you find a solution to the problem :-)

[github-actions[bot]]

This issue has been automatically marked as stale because it has not had recent activity. If you think this still needs to be addressed please comment on this thread.

Please note that issues that do not follow the contributing guidelines are likely to be ignored.