1 安装huggingface cli

使用以下命令安装HuggingFace CLI

注意:huggingface_hub 依赖于 Python>=3.8,此外需要安装 0.17.0 及以上的版本,推荐0.19.0+。

pip install -U huggingface_hub

注意,在Windows系统上需要以管理员权限运行终端工具,不然就算安装完成也不能使用huggingface-cil命令,会出现以下错误

huggingface-cli 不是内部或者外部命令,也不是可执行程序

2 设置国内镜像源

Linux:

export HF_ENDPOINT="https://hf-mirror.com"

Windows:

$env:HF_ENDPOINT = "https://hf-mirror.com"

3 使用huggingface cli下载模型

以facebook的sapiens为例,链接:https://huggingface.co/facebook/sapiens/tree/main

可以使用以下命令下载模型

huggingface-cli download --resume-download facebook/sapiens --local-dir ./sapiens_model

如果需要下载数据集,则可以使用以下命令

huggingface-cli download --resume-download --repo-type dataset lavita/medical-qa-shared-task-v1-toy