尝试本地加载yi1.5-6b模型,报错 OSError: Error no file named pytorch_model.bin, tf_model.h5, model.ckpt.index or flax_model.msgpack
网上查询后多数认为是缺失文件model.safetensors.index.json,但实际上我的文件夹中已包含此文件。根据huggingface上的模型文件结构
推测是因为 transformers 不能正确识别这种safetensors模型文件,于是查询了yi1.5模型的官方github--requirements
先升级了transformers包,果然不报错了。再根据报错进一步升级了其他包。
附一个其他错误:ImportError: cannot import name 'split_torch_state_dict_into_shards' from 'huggingface_hub' (……)
解决方法:pip install --upgrade huggingface_hub
参考:https://github.com/TencentARC/InstantMesh/issues/116
使用模型前还是要多看看官方给的配置需求 (˘•ω•˘)