如何从huggingface下载模型？

Question

https://huggingface.co/models

比如我想下载'bert-base-uncased'，但是找不到'Download'link。请帮忙。还是不能下载？

Answer 1

首次使用时，模型会自动缓存到本地。所以，要下载一个模型，你所要做的就是运行model card中提供的代码（我选择了bert-base-uncased对应的模型卡）。

在页面的右上角，您可以找到一个名为“Use in Transformers”的按钮，它甚至还为您提供了示例代码，向您展示了如何在 Python 中使用它。同样，对于 bert-base-uncased，这会为您提供以下代码片段：

from transformers import AutoTokenizer, AutoModelForMaskedLM
  
tokenizer = AutoTokenizer.from_pretrained("bert-base-uncased")
model = AutoModelForMaskedLM.from_pretrained("bert-base-uncased")

当您第一次运行此代码时，您会看到屏幕上出现一个下载栏。如果您想找到 Huggingface 存储模型的实际文件夹，请参阅（免责声明：我给出了其中一个答案）。

Answer 2

接受的答案很好，但是编写代码下载模型并不总是很方便。从 huggingface 获取模型似乎 git 效果很好。这是一个例子：

git lfs clone https://huggingface.co/sberbank-ai/ruT5-base

其中 'lfs' 停留 'large file storage'。从技术上讲，这个命令已被弃用，简单的 'git clone' 应该可以工作，但是你需要设置过滤器以不跳过大文件 ()

Answer 3

我同意 Jahjajaka 的回答。此外，您可以通过单击名为“Use in Transformers”的按钮找到 git url，如图所示。

如何从huggingface下载模型？

How to download model from huggingface?

transformer

huggingface-transformers