如何从拥抱脸下载模型？

Posted 2023-03-04

【中文标题】如何从拥抱脸下载模型？

【英文标题】：How to download model from huggingface?

【发布时间】：2021-08-08 05:44:18

【问题描述】：

https://huggingface.co/models

例如，我想下载“bert-base-uncased”，但找不到“下载”链接。请帮忙。还是不能下载？

【参考方案1】：

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

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

“lfs”用于“大文件存储”的位置。从技术上讲，此命令已弃用，简单的“git clone”应该可以工作，但是您需要设置过滤器以不跳过大文件 (How do I clone a repository that includes Git LFS files?)

【参考方案2】：

当您第一次使用模型时，模型会自动在本地缓存。 因此，要下载模型，您只需运行model card 中提供的代码（我为bert-base-uncased 选择了相应的模型卡）。

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

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

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

【讨论】：

如何将模型下载到本地？

如果我想将模型下载到特定目录怎么办？

我认为这值得自己提出问题；事实上，我相信这里给出了一个可能的答案：***.com/questions/63312859/…

【参考方案3】：

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

【讨论】：

这应该是对 Jahjajaka 的回答的评论

我无权在 jahjajaka 的回答下发表评论，至少需要 50 个声望。