下载代码

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'])

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