从 Hugging Face 下载 Stable Diffusion 模型到本地

1,014 阅读2分钟

要从 Hugging Face 下载 Stable Diffusion 模型到本地,您可以遵循以下步骤:

步骤 1: 安装 Hugging Face Hub 库

确保您的环境中安装了 huggingface_hub 库。如果没有安装,可以通过以下命令安装:

bashCopy code
pip install huggingface_hub

步骤 2: 使用 Hugging Face Hub API 下载模型

Hugging Face Hub 提供了一个 API,允许您通过编程方式下载模型。以下是一个简单的 Python 脚本示例,展示了如何下载 Stable Diffusion 模型:

pythonCopy code
from huggingface_hub import hf_hub_download

# 指定模型的仓库名称
repo_name = "CompVis/stable-diffusion-v1-4"

# 指定要下载的文件名。这里需要确切知道模型的文件名
# 例如: "pytorch_model.bin", "config.json" 等
file_names = ["file_name_1", "file_name_2", ...]

for file_name in file_names:
    # 下载文件到本地
    file_path = hf_hub_download(repo_id=repo_name, filename=file_name)

    print(f"Downloaded {file_name} to {file_path}")

在这个脚本中,您需要替换 file_names 列表中的文件名为 Stable Diffusion 模型实际的文件名。这些文件可能包括模型的权重文件、配置文件等。

步骤 3: 确定要下载的文件

您需要知道 Stable Diffusion 模型的确切文件名。通常,这些信息可以在模型的 Hugging Face 页面上找到。对于大多数模型,至少需要下载以下两类文件:

  • 模型权重文件(如 .bin.pt 文件)
  • 配置文件(如 .json 文件)

步骤 4: 加载模型

一旦您下载了所有必要的文件,可以使用适当的 Python 库(如 transformerstorch 或其他)加载并使用这些本地文件。

注意事项

  • 确保您知道要下载的确切文件名。
  • 某些模型可能需要特定版本的库或额外的依赖项,因此请参考模型页面上的说明。
  • 如果您不确定哪些文件是必需的,或者如何加载下载的模型,建议查看 Hugging Face 模型页面上的文档和使用说明。