使用 huggingface-cli 下载模型

简介

在进行深度学习项目时，下载大规模的预训练模型是常见的需求。huggingface-cli 是 Hugging Face 提供的命令行工具，用于下载和管理模型。

如果模型较大，或者需要在后台运行下载任务时，可以结合 nohup 命令来确保下载过程不会被中断。本文将介绍如何使用这些工具进行模型下载，并如何查看下载进程。

最新完整教程可以参考链接，Huggingface-Command Line Interface (CLI)

hf-cli 使用

huggingface-cli 的安装

使用下面的命令进行安装：

安装完毕之后，可以使用下面的命令查看是否安装成功：

使用 huggingface-cli 下载模型

安装完毕之后，我们可以通过简单的命令下载 Hugging Face 上的模型。例如，我们可以使用以下命令下载 sd-controlnet-seg 模型：

上面的命令的含义为：

• • nohup：使得下载命令在后台运行，并且即使终端关闭，下载进程依然会继续执行。
  • huggingface-cli download lllyasviel/sd-controlnet-seg：指定下载模型的名称（此处是 `sd-controlnet-seg`）。
  • --local-dir="/home/sd-controlnet-seg"：指定模型存储的本地目录。
  • > download.log 2>&1 &：将下载过程中的标准输出和错误信息重定向到 download.log 文件，并将进程放到后台执行。

同样地，你可以使用类似的命令来下载其他模型。例如：

• hf, sd-controlnet-canny
• hf, stable-diffusion-xl-base-1.0

使用 ps 和 pgrep 查看下载进程

为了查看下载进程是否正在运行，可以使用 ps 命令查看系统中所有的进程，并筛选出涉及 huggingface-cli 的进程：

如果希望使用更简洁的方式，也可以使用 pgrep 命令来查看是否存在 huggingface-cli 相关的进程：

这两个命令会返回包含 huggingface-cli 关键字的进程信息。如果下载进程正在运行，你将能够看到对应的进程 ID 和其他相关信息。