下载代码
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-chinesesnapshots38fda776740d17609554e879e3ac7b9837bdb5ee")
model = AutoModelForMaskedLM.from_pretrained("bert-base-chinese")
en2zh = pipeline("fill-mask", model=model, tokenizer=tokenizer)
print(en2zh("不要[MASK]我。")[0]['sequence'])
最后
以上就是醉熏大船最近收集整理的关于transformers下载模型并引用下载代码引用模型的全部内容，更多相关transformers下载模型并引用下载代码引用模型内容请搜索靠谱客的其他文章。
本图文内容来源于网友提供，作为学习参考使用，或来自网络收集整理，版权属于原作者所有。
发表评论 取消回复