python下载HuggingFace的模型_python 代码下载huggingface模型

首先第一步，在linux上开启魔法、梯子

验证是否开启魔法

curl --proxy http://127.0.0.1:7890 https://www.google.com

如果没有返回谷歌页面的值，说明魔法开启失败。

第二步

proxies = {

    'http': 'http://127.0.0.1:7890/',

    'https': 'http://127.0.0.1:7890/'

}

from transformers import AutoTokenizer, AutoModelForSequenceClassification

model_name = "nlptown/bert-base-multilingual-uncased-sentiment"

model = AutoModelForSequenceClassification.from_pretrained(model_name,proxies=proxies)

tokenizer = AutoTokenizer.from_pretrained(model_name,proxies=proxies)

 在代码中开启代理

就可以正常下载啦。

方法二：

如果没有科学上网环境，可以使用国内的网站提供权重

https://modelscope.cn/models/AI-ModelScope/https://modelscope.cn/models/AI-ModelScope/