大模型部署避坑指南--OSError: Unable to load weights from pytorch checkpoint file for
If you tried to load a PyTorch model from a TF 2.0 checkpoint, please set from_tf=True.
·
问题描述：
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
魔乐社区（Modelers.cn) 是一个中立、公益的人工智能社区，提供人工智能工具、模型、数据的托管、展示与应用协同服务，为人工智能开发及爱好者搭建开放的学习交流平台。社区通过理事会方式运作，由全产业链共同建设、共同运营、共同享有，推动国产AI生态繁荣发展。
更多推荐
3
0- 0
已为社区贡献7条内容
所有评论(0)