google

/

ncsnpp-church-256

like 2

Follow

Google 4,591

Convert to a diffusers-compatible format.

Here you can find the script.

I used that script to convert the original ncscnpp cifar10 and found this error during conversion. can you help me how to solve this 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