使用提供的模型权重推理时报模型加载出错

[silencewa]

Traceback (most recent call last):
File "web_demo.py", line 154, in
main(args)
File "web_demo.py", line 90, in main
model, model_args = AutoModel.from_pretrained(
File "/root/miniconda3/lib/python3.8/site-packages/sat/model/base_model.py", line 275, in from_pretrained
load_checkpoint(model, args, load_path=model_path, prefix=prefix)
File "/root/miniconda3/lib/python3.8/site-packages/sat/training/model_io.py", line 222, in load_checkpoint
sd = torch.load(checkpoint_name, map_location='cpu')
File "/root/miniconda3/lib/python3.8/site-packages/torch/serialization.py", line 713, in load
return _legacy_load(opened_file, map_location, pickle_module, **pickle_load_args)
File "/root/miniconda3/lib/python3.8/site-packages/torch/serialization.py", line 920, in _legacy_load
magic_number = pickle_module.load(f, **pickle_load_args)
_pickle.UnpicklingError: invalid load key, 'v'.