Shape mismatch in load.state_dict()

[santurini]

I am trying to running demo.py on some personalized prompts but I am not able to load the model as I get the following error:

RuntimeError: Error(s) in loading state_dict for MovementConvEncoder:
	size mismatch for main.0.weight: copying a param with shape torch.Size([512, 259, 4]) from checkpoint, the shape in current model is torch.Size([512, 18, 4]).

And the command I am using is:

python demo.py --cfg ./configs/config_h3d_stage3.yaml --example ./demos/t2m.txt