Unexpected key `base_sizes_/454056960/

[wangxihao]

torch.load(motion_module_path, map_location="cpu")
*** RuntimeError: Unexpected key base_sizes_/454056960/ passed to setTensorMetadata.

[yangshiyu89]

torch.load(motion_module_path, map_location="cpu") *** RuntimeError: Unexpected key base_sizes_/454056960/ passed to setTensorMetadata.

+1, when I test the basic torch.load funs, this issue happened to me too.

import torch
motion_module_path = 'resources/models/fancyvideo_ckpts/vae_3d_61_frames/mp_rank_00_model_states.pt'
state_dict = torch.load(motion_module_path, map_location="cpu")['module']

[MaAo]

Hi, @wangxihao @yangshiyu89 thanks for your attention to our work, this problem may be a problem with the downloaded model, here is an alternate download address in addition to huggingface.

wget -O resources/models/fancyvideo_ckpts.zip "https://drive.google.com/uc?export=download&id=1m4UqKVQ3POI5ei1A9yppHX_H--8PKMtn"

[yangshiyu89]

Thank you @MaAo for your reply. I found the new version of the model did not work well either.
Finally, I have tested different versions of torch. The version of 2.1.2 from requirements.txt runs well, but 2.0.x failed.