背景
加载torch的huggingface的bin模型,未使用复杂处理逻辑,但是出现异常,详细报错内容
OSError: Unable to load weights from pytorch checkpoint file for
'xxxxx/huggingface_model/bart-large-chinese'
at 'xxxxxx/huggingface_model/bart-large-chinese/pytorch_model.bin'
If you tried to load a PyTorch model from a TF 2.0 checkpoint, please set from_tf=True.
尝试
- 设置from_tf=True, 结果并有改善
- 按照提示,以为是TF的版本太高,于是查看TF版本,结果报错
ImportError: /usr/lib64/libstdc++.so.6: version `GLIBCXX_3.4.20' not found
(required by xxxxx/lib/python3.7/site-packages/tensorflow/python/_pywrap_tensorflow_internal.so)
发现机器上并没有安装TF
解决办法
说来解决办法有点水,参考之前有效的机器,发现torch的版本是1.11,后来将原始版本升级到1.11