赞
踩
(1) 安装依赖
pip install -U huggingface_hub hf_transfer
可使用清华源:
pip install -U huggingface_hub hf_transfer
-i https://pypi.tuna.tsinghua.edu.cn/simple
(2) 命令
设置镜像站
export HF_ENDPOINT=https://hf-mirror.com
开启hf_transfer加速下载
export HF_HUB_ENABLE_HF_TRANSFER=1
下载命令
huggingface-cli download --resume-download 模型名称 --local-dir 本地目录
以"openai-community/gpt2-medium"为例:
huggingface-cli download --resume-download openai-community/gpt2-medium
如果不设置 --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 本地目录
pip install -U huggingface_hub
命令后解决pip install chardet
将 huggingface.co 直接替换为 hf-mirror.com,使用浏览器或者 wget -c、curl -L、aria2c 等命令行方式。下载需登录的模型需命令行添加 --header hf_*** 参数。
参考文章:
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。