背景
使用hugging face 中的transformers库无法加载预训练模型文件
具体代码和报错
使用以下代码的时候,发生报错:
from transformers import AutoTokenizer, AutoModelForMaskedLM
tokenizer = AutoTokenizer.from_pretrained("hfl/chinese-macbert-base")
model = AutoModelForMaskedLM.from_pretrained("hfl/chinese-macbert-base")
报错:
OSError: Can't load config for 'hfl/chinese-macbert-base'. If you were trying
无法连接到服务器,没法下载权重文件
解决方法
手动下载 需要的权重文件和配置文件,然后再次加载