Error loading model trained using previous version.

[hizzlehoff]

I'm getting the following error since the update from 2 days ago when loading a previously trained model:

I can keep working by using the previous version:
"!pip install denoising_diffusion_pytorch==0.31"

Thank you.

/usr/local/lib/python3.7/dist-packages/torch/nn/modules/module.py in load_state_dict(self, state_dict, strict)
1603 if len(error_msgs) > 0:
1604 raise RuntimeError('Error(s) in loading state_dict for {}:\n\t{}'.format(
-> 1605 self.class.name, "\n\t".join(error_msgs)))
1606 return _IncompatibleKeys(missing_keys, unexpected_keys)
1607

RuntimeError: Error(s) in loading state_dict for GaussianDiffusion:
Missing key(s) in state_dict: "model.downs.0.3.1.weight", "model.downs.0.3.1.bias", "model.downs.1.3.1.weight", "model.downs.1.3.1.bias", "model.downs.2.3.1.weight", "model.downs.2.3.1.bias".
Unexpected key(s) in state_dict: "model.downs.0.3.weight", "model.downs.0.3.bias", "model.downs.1.3.weight", "model.downs.1.3.bias", "model.downs.2.3.weight", "model.downs.2.3.bias".