配置、使用transformers包-----下载hugging_face中的预训练模型到本地使用
1. 访问 https://huggingface.co/models 网站,输入我们想要的预训练模型

2. 下载需要的config.json文件、vocab.txt文件和对应的.bin文件
3. 在项目中新建一个文件夹models,将刚才下载的文件放到文件夹models中
4. 在代码中使用下载的预训练模型,使用的路径是存放了所有文件的文件夹路径,注意添加local_files_only=True
from transformers import BertModel
model = BertModel.from_pretrained('./models',local_files_only=True)
本文介绍了如何从HuggingFace库下载预训练模型(如BERT),包括获取config.json、vocab.txt和.bin文件,以及如何在项目中正确配置并使用这些模型。
1037

被折叠的 条评论
为什么被折叠?



