下载代码
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'])