背景

使用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

无法连接到服务器,没法下载权重文件

解决方法

手动下载 需要的权重文件和配置文件,然后再次加载

1. 访问 https://huggingface.co/models 网站,输入我们想要的与训练模型

在这里插入图片描述

2. 找到我们的需要的权重文件 .bin 文件 和 配置文件 config.json ,点击下载

在这里插入图片描述

在这里插入图片描述

3. 将我们下载的文件,放入到指定的文件夹中

在这里插入图片描述

3. 再次运行,查看情况

在这里插入图片描述

Logo

旨在为数千万中国开发者提供一个无缝且高效的云端环境,以支持学习、使用和贡献开源项目。

更多推荐