如何从huggingface下载模型？

Question

例如，我想在https://huggingface.co/modelsbert-base-uncased上下载，但找不到“下载”链接。或者说不能下载？

Answer 1

git clone从 Huggingface 获取模型效果很好。这是一个例子：

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

git lfs请注意，如果您想实际下载存储库中的大文件，而不仅仅是对它们的引用，则需要安装。如果您运行git lfs --version并收到“命令无法识别”消息，则说明您尚未安装它。您可以从官方 git-lfs 网站获取最新版本，或使用包管理器安装旧版本 - 例如，在 Ubuntu 上：

apt-get install git-lfs
git lfs install

相关：如何克隆包含 Git LFS 文件的存储库？

Answer 2

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

在页面的右上角，您可以找到一个名为“在 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 3

hf_hub_download从图书馆使用怎么样huggingface_hub？

hf_hub_download返回下载模型的本地路径，以便您可以使用另一个 shell 命令来挂钩这个衬垫。

python3 -c 'from huggingface_hub import hf_hub_download; downloaded_model_path = hf_hub_download(
                                                          repo_id="CompVis/stable-diffusion-v-1-4-original",
                                                          filename="sd-v1-4.ckpt",
                                                          use_auth_token=True
                                                         ); print(downloaded_model_path)'

Answer 4

要从 Hugging Face 下载模型，您可以使用官方 CLI 工具或库中的huggingface-cliPython 方法。snapshot_downloadhuggingface_hub

使用huggingface-cli：

要下载“bert-base-uncased”模型，只需运行：

$ huggingface-cli download bert-base-uncased

snapshot_download在Python中使用：

from huggingface_hub import snapshot_download

snapshot_download(repo_id="bert-base-uncased")

这些工具使从 Hugging Face 模型中心下载模型变得快速、轻松。

更多信息和高级使用，可以参考 Hugging Face 官方文档：

• huggingface-cli文档

• snapshot_download文档