下载Hugging Face模型到本地并使用

打开 Hugging Face 主页：

Hugging Face – The AI community building the future.

搜索自己需要的哪个模型名字即可，这里我以 facebook/bart-base 做为例子。

输入回车如果找到了，就会自动跳转到模型的页面：

可以看 How to use 看如何使用该模型，当然了，我们是无法直连的，所以这里把模型需要的模型下载到本地去使用，点击 File and versions，根据自己需要下载即可：

通常，Pytorch 需要下载如下内容：

TensorFlow 需要下载如下内容：

Pytorch 用的是 pytorch_model.bin，TensorFlow 用的是 tf_model.h5，json 配置文件一般都下载，反正也没多大，不然缺少文件还需要重新下载。

词汇表名字不固定，可能是 vocab.txt 或者 merges.txt。

下载完之后，在自己的项目里创建文件夹，把下载的东西放进去。

最后使用 transformers 库进行调用即可。

from transformers import BertModel,BertTokenizer

BERT_PATH = '/bert_model'

tokenizer = BertTokenizer.from_pretrained(BERT_PATH)
demo_data = "story of a man who has unnatural feelings"
encoded_inputs = tokenizer(demo_data, padding=True, truncation=True, return_tensors='pt', max_length=512)

model = BertModel.from_pretrained(BERT_PATH)
outputs = model(**encoded_inputs)
print(outputs)