Happy New Year; model_split KeyError

[GloryyrolG]

hi @ALL @Annbless @seaman1900 @AK391 ,

may i ask how to solve the issue of the model_split.py? thx & best, happy new year,

> python tools/model_split.py --source ckpt/pose2d/vitpose+_huge_wholebody.pth

Traceback (most recent call last):
  File "tools/model_split.py", line 104, in <module>
    main()
  File "tools/model_split.py", line 85, in main
    new_ckpt['state_dict'][tensor_name] = new_ckpt['state_dict'][tensor_name.replace('keypoint_head', f'associate_keypoint_heads.{i}')]
KeyError: 'associate_keypoint_heads.0.deconv_layers.0.weight'