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

[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'.