当前位置:   article > 正文

Huggingface镜像站使用及常见报错_hugging镜像

hugging镜像

方法一:使用 huggingface-cli 下载模型

(1) 安装依赖

pip install -U huggingface_hub hf_transfer
  • 1

可使用清华源

pip install -U huggingface_hub hf_transfer
 -i https://pypi.tuna.tsinghua.edu.cn/simple
  • 1
  • 2

(2) 命令
设置镜像站

export HF_ENDPOINT=https://hf-mirror.com
  • 1

开启hf_transfer加速下载

export HF_HUB_ENABLE_HF_TRANSFER=1
  • 1

下载命令

huggingface-cli download --resume-download 模型名称 --local-dir 本地目录
  • 1

以"openai-community/gpt2-medium"为例:

huggingface-cli download --resume-download openai-community/gpt2-medium
  • 1

如果不设置 --local-dir,模型会默认保存到.cache/huggingface/hub/models--openai-community--gpt2-medium/
(3) 下载需要登录的模型(Gated Model)
添加–token hf_***参数,其中hf_***是 access token,在huggingface官网获取

huggingface-cli download --token hf_*** --resume-download 模型名称 --local-dir 本地目录
  • 1

常见报错

  1. huggingface-cli: error: invalid choice: ‘download’ (choose from ‘env’, ‘login’, ‘whoami’, ‘logout’, ‘repo’, ‘lfs-enable-largefiles’, ‘lfs-multipart-upload’, ‘scan-cache’, ‘delete-cache’)
    • 可能是由于python版本<3.8,下载的huggingface-cli版本过低;在python>=3.8环境下运行pip install -U huggingface_hub命令后解决
  2. ModuleNotFoundError: No module named ‘chardet’
    • 安装对应模块即可:pip install chardet

方法二:使用url直接下载

将 huggingface.co 直接替换为 hf-mirror.com,使用浏览器或者 wget -c、curl -L、aria2c 等命令行方式。下载需登录的模型需命令行添加 --header hf_*** 参数。

参考文章:

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/小小林熬夜学编程/article/detail/264134
推荐阅读
相关标签
  

闽ICP备14008679号