transformers下载模型并引用

最新推荐文章于 2024-04-30 23:23:05 发布

钱甫新

最新推荐文章于 2024-04-30 23:23:05 发布

阅读量1.3k

点赞数

分类专栏：人工智能文章标签：深度学习人工智能

本文链接：https://blog.csdn.net/Mr_Qian_Ives/article/details/126748789

版权

人工智能专栏收录该内容

2 篇文章 0 订阅

订阅专栏

下载代码

from huggingface_hub import snapshot_download
# 下载模型到当前目录下
snapshot_download(repo_id="bert-base-chinese",cache_dir=".")

用cmd以管理员身份运行代码（没有管理员身份会报错）
在这里插入图片描述

引用模型

在这里插入图片描述

from transformers import AutoTokenizer, AutoModelForMaskedLM, pipeline

tokenizer = AutoTokenizer.from_pretrained(pretrained_model_name_or_path=r"models--bert-base-chinese\snapshots\38fda776740d17609554e879e3ac7b9837bdb5ee")

model = AutoModelForMaskedLM.from_pretrained("bert-base-chinese")
en2zh = pipeline("fill-mask", model=model, tokenizer=tokenizer)

print(en2zh("不要[MASK]我。")[0]['sequence'])