【问题标题】:How to download model from huggingface?如何从拥抱脸下载模型?
【发布时间】:2021-08-08 05:44:18
【问题描述】:
https://huggingface.co/models

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

【问题讨论】:

    标签: huggingface-transformers transformer


    【解决方案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(免责声明:我给出了其中一个答案)。

      【讨论】:

      • 如何将模型下载到本地?
      • 如果我想将模型下载到特定目录怎么办?
      • 我认为这值得自己提出问题;事实上,我相信这里给出了一个可能的答案:stackoverflow.com/questions/63312859/…
      【解决方案3】:

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

      【讨论】:

      • 这应该是对 Jahjajaka 的回答的评论
      • 我无权在 jahjajaka 的回答下发表评论,至少需要 50 个声望。
      猜你喜欢
      • 2022-01-17
      • 2020-11-06
      • 2022-01-04
      • 1970-01-01
      • 2021-01-15
      • 2021-12-30
      • 2021-08-16
      • 2021-11-02
      • 2021-12-21
      相关资源
      最近更新 更多