大模型部署避坑指南--OSError: Unable to load weights from pytorch checkpoint file for_pytorch_model-00001-of-00002.bin

作者：不正经 | 2024-04-09 10:09:51

踩

pytorch_model-00001-of-00002.bin

问题描述：

OSError: Unable to load weights from pytorch checkpoint file for
'/home/aistudio/vicuna-7b-delta-v0/pytorch_model-00001-of-00002.bin' at
'/home/aistudio/vicuna-7b-delta-v0/pytorch_model-00001-of-00002.bin'. If you tried to load a PyTorch model from a TF 2.0 checkpoint, please set from_tf=True.

产生原因：

torch 版本低

解决办法：

pip install --upgrade torch torchvision

声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/不正经/article/detail/391995