Stable Diffusion2模型下载

### 下载 Stable Diffusion 2 模型文件

为了下载并使用 Stable Diffusion 2 模型，可以采用如下方法：

通过 `diffusers` 库来加载特定版本的 Stable Diffusion 模型。对于 Stable Diffusion 2 版本而言，可以通过指定预训练权重名称来进行加载[^1]。

from diffusers import StableDiffusionPipeline, EulerAncestralDiscreteScheduler
import torch

model_id = "stabilityai/stable-diffusion-2"

# 使用Euler Ancestral采样器初始化调度程序
scheduler = EulerAncestralDiscreteScheduler.from_pretrained(model_id, subfolder="scheduler")

# 创建管道实例
pipe = StableDiffusionPipeline.from_pretrained(
    model_id,
    scheduler=scheduler,
    revision="fp16",
    torch_dtype=torch.float16,
)

# 将模型移动到GPU上运行（如果有可用的话）
device = "cuda" if torch.cuda.is_available() else "cpu"
pipe.to(device)

这段代码展示了如何利用 Hugging Face 的 `diffusers` 库来获取 Stable Diffusion 2 并设置其工作环境。这里选择了 FP16 数据类型以提高性能，并指定了 `"stabilityai/stable-diffusion-2"` 这一 ID 来确保正确加载目标版本的模型[^2]。