美文网首页
hugging face预训练模型本地读取指南

hugging face预训练模型本地读取指南

作者: 不可能打工 | 来源:发表于2023-05-24 16:59 被阅读0次

    使用Hugging Face的Transformers库来下载预训练模型bert模型时。以下是下载和使用该模型的示例代码:

    from transformers import BertTokenizer, BertModel
    
    tokenizer = BertTokenizer.from_pretrained('bert-base-chinese')
    model = BertModel.from_pretrained('bert-base-chinese')
    

    这将自动下载预训练模型bert-base-chinese并创建一个BertTokenizer和BertModel对象,您可以使用它们来进行文本编码和模型推理。

    这一切的前提,建立在开发机是可以联网的

    当机器无法联网时,读取预训练模型会失败,此时需要我们在能上网的地方下载模型文件,上传开发机。然后从本地缓存中加载模型。

    方法如下

    首先进入hugging face,找到自己想要下载的模型,点击下载对应的文件。针对pytorch模型,需要下载以下文件,下载后新建一个文件夹bert-base-chinese,将这些文件放入,一并上传开发机。


    image.png

    修改读取预训练模型代码如下,即可正常运行。

    config = BertConfig.from_json_file("bert-base-chinese/config.json")
    model = BertModel.from_pretrained("bert-base-chinese/pytorch_model.bin", config=config)
    tokenizer = BertTokenizer.from_pretrained('bert-base-chinese/') ##注意此处为本地文件夹
    

    相关文章

      网友评论

          本文标题:hugging face预训练模型本地读取指南

          本文链接:https://www.haomeiwen.com/subject/alfqedtx.html