I'm encountering issues loading model checkpoints, resulting in multiple size mismatch errors. Here is the context and the errors encountered:
Problem
I attempted to load the model checkpoint. However, there are numerous size mismatches in the parameters when trying to load from the upgraded checkpoints.
Errors:
RuntimeError: Error(s) in loading state_dict for DiffusionPipeline:
size mismatch for latent_embedder.out_enc.0.conv.weight: copying a param with shape torch.Size([8, 512, 3, 3]) from checkpoint, the shape in current model is torch.Size([16, 512, 3, 3]).
size mismatch for latent_embedder.out_enc.0.conv.bias: copying a param with shape torch.Size([8]) from checkpoint, the shape in current model is torch.Size([16]).
...
Could you provide guidance on how to properly address these mismatches?
I'm encountering issues loading model checkpoints, resulting in multiple
size mismatch
errors. Here is the context and the errors encountered:Problem I attempted to load the model checkpoint. However, there are numerous size mismatches in the parameters when trying to load from the upgraded checkpoints.
Errors:
Could you provide guidance on how to properly address these mismatches?
Thank you!