在hugging face中下载一个模型,经常失败,显示网络连接错误。

只需要在开头加入

import os
os.environ["HF_ENDPOINT"] = "https://hf-mirror.com"即可。
import os
os.environ["HF_ENDPOINT"] = "https://hf-mirror.com"
from transformers import AutoTokenizer, AutoModelForTokenClassification
# 加载预训练的tokenizer
tokenizer = AutoTokenizer.from_pretrained("Clinical-AI-Apollo/Medical-NER")
# 加载预训练的模型
model = AutoModelForTokenClassification.from_pretrained("Clinical-AI-Apollo/Medical-NER")
inputs = tokenizer("45 year old woman diagnosed with CAD", return_tensors="pt")
outputs = model(**inputs)

Logo

魔乐社区(Modelers.cn) 是一个中立、公益的人工智能社区,提供人工智能工具、模型、数据的托管、展示与应用协同服务,为人工智能开发及爱好者搭建开放的学习交流平台。社区通过理事会方式运作,由全产业链共同建设、共同运营、共同享有,推动国产AI生态繁荣发展。

更多推荐