基本下载命令
huggingface-cli download <模型名称> [文件名]
<模型名称>
:Hugging Face Hub 上的模型仓库名称,例如bert-base-uncased
。[文件名]
:(可选)指定要下载的文件名称,例如config.json
。如果不指定文件名,则会下载整个模型仓库。
常用参数
-
指定下载路径
huggingface-cli download <模型名称> --local-dir <本地路径>
--local-dir
:指定下载文件的本地目录。如果没有指定,默认会下载到系统的缓存目录中。
-
启用断点续传
huggingface-cli download <模型名称> --resume-download
--resume-download
:如果下载中断,可以重新开始下载,而不会从头开始。
-
下载特定版本的文件
huggingface-cli download <模型名称> <文件名> --revision <版本号>
--revision
:指定要下载的模型版本号。如果不指定,默认下载最新版本。
-
启用多线程下载(加速下载)
huggingface-cli download <模型名称> --num-procs <线程数>
--num-procs
:指定下载时使用的线程数。默认为单线程,增加线程数可以加速下载。
-
下载整个模型仓库
huggingface-cli download <模型名称>
如果不指定文件名,
huggingface-cli
会自动下载模型仓库中的所有文件。
示例命令
以下是一些具体的示例命令:
-
下载整个模型仓库到默认缓存目录
huggingface-cli download bert-base-uncased
-
下载指定文件到默认缓存目录
huggingface-cli download bert-base-uncased config.json
-
下载整个模型仓库到指定目录
huggingface-cli download bert-base-uncased --local-dir ./models
-
下载指定文件到指定目录
huggingface-cli download bert-base-uncased config.json --local-dir ./models
-
下载特定版本的文件
huggingface-cli download bert-base-uncased config.json --revision v1.0.0
-
启用断点续传下载
huggingface-cli download bert-base-uncased --resume-download
-
启用多线程下载
huggingface-cli download bert-base-uncased --num-procs 4
其他注意事项
- 登录账号:如果需要下载私有模型或上传模型,需要先登录 Hugging Face 账号:
huggingface-cli login
- 配置国内镜像:国内用户可以配置 Hugging Face 的国内镜像源以加速下载:
或在 Windows 中:export HF_ENDPOINT=https://hf-mirror.com
$env:HF_ENDPOINT = "https://hf-mirror.com"
通过以上指令和参数,你可以灵活地从 Hugging Face Hub 下载模型和文件。