模型简介
模型特点
模型能力
使用案例
🚀 Stable Diffusion v1 - 5模型卡片
Stable Diffusion是一种潜在文本到图像的扩散模型,能够根据任何文本输入生成逼真的图像。它为图像生成领域带来了新的可能性,可广泛应用于艺术创作、设计等多个领域。
🚀 快速开始
Stable Diffusion是一种潜在文本到图像的扩散模型,能够根据任何文本输入生成逼真的图像。如需了解有关Stable Diffusion工作原理的更多信息,请查看 🤗的Stable Diffusion博客。
你可以使用 🧨Diffusers库 或 RunwayML GitHub仓库 来使用此模型。
💻 使用示例
基础用法
from diffusers import StableDiffusionPipeline
import torch
model_id = "runwayml/stable-diffusion-v1-5"
pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16)
pipe = pipe.to("cuda")
prompt = "a photo of an astronaut riding a horse on mars"
image = pipe(prompt).images[0]
image.save("astronaut_rides_horse.png")
如需更详细的说明、用例和JAX中的示例,请遵循 此处 的说明。
原始GitHub仓库
- 下载权重
- v1 - 5 - pruned - emaonly.ckpt - 4.27GB,仅EMA权重。使用较少的VRAM,适合推理。
- v1 - 5 - pruned.ckpt - 7.7GB,EMA + 非EMA权重。使用更多的VRAM,适合微调。
- 遵循 此处 的说明。
✨ 主要特性
- 文本驱动图像生成:能够根据文本输入生成逼真的图像。
- 多平台支持:可以与Diffusers库和RunwayML GitHub仓库配合使用。
📚 详细文档
模型详情
属性 | 详情 |
---|---|
开发者 | Robin Rombach, Patrick Esser |
模型类型 | 基于扩散的文本到图像生成模型 |
语言 | 英语 |
许可证 | CreativeML OpenRAIL M许可证 是一种 Open RAIL M许可证,改编自 BigScience 和 RAIL Initiative 在负责任AI许可领域的联合工作。另见 关于BLOOM Open RAIL许可证的文章,本许可证基于该文章。 |
模型描述 | 这是一个可用于根据文本提示生成和修改图像的模型。它是一个 潜在扩散模型,使用固定的预训练文本编码器 (CLIP ViT - L/14),如 Imagen论文 中所建议。 |
更多信息资源 | GitHub仓库,论文 |
引用格式 | @InProceedings{Rombach_2022_CVPR, |
使用方式
直接使用
该模型仅用于研究目的。可能的研究领域和任务包括:
- 安全部署有可能生成有害内容的模型。
- 探索和理解生成模型的局限性和偏差。
- 艺术作品的生成以及在设计和其他艺术过程中的应用。
- 在教育或创意工具中的应用。
- 生成模型的研究。
排除的使用方式如下所述。
滥用、恶意使用和超出范围的使用
注意:本节内容取自 [DALLE - MINI模型卡片](https://huggingface.co/dalle - mini/dalle - mini),但同样适用于Stable Diffusion v1。
该模型不应被用于故意创建或传播为人们创造敌对或疏离环境的图像。这包括生成人们可预见会感到不安、痛苦或冒犯的图像;或传播历史或当前刻板印象的内容。
超出范围的使用
该模型并非用于对人物或事件进行事实或真实的呈现,因此使用该模型生成此类内容超出了该模型的能力范围。
滥用和恶意使用
使用该模型生成对个人残忍的内容是对该模型的滥用。这包括但不限于:
- 生成贬低、非人化或以其他方式伤害人们或其环境、文化、宗教等的表现形式。
- 故意推广或传播歧视性内容或有害刻板印象。
- 未经个人同意冒充他人。
- 未经可能看到的人的同意生成色情内容。
- 错误信息和虚假信息。
- 严重暴力和血腥场面的表现形式。
- 违反使用条款分享受版权保护或有许可的材料。
- 违反使用条款分享受版权保护或有许可材料的修改版本。
局限性和偏差
局限性
- 模型无法实现完美的照片写实主义。
- 模型无法渲染清晰可读的文本。
- 模型在涉及组合性的更困难任务上表现不佳,例如渲染与“红色立方体在蓝色球体之上”对应的图像。
- 面部和人物总体上可能无法正确生成。
- 模型主要使用英语字幕进行训练,在其他语言中的效果不佳。
- 模型的自动编码部分存在信息损失。
- 模型在大规模数据集 [LAION - 5B](https://laion.ai/blog/laion - 5b/) 上进行训练,该数据集包含成人内容,在没有额外安全机制和考虑的情况下不适合产品使用。
- 未使用额外措施对数据集进行去重。因此,我们观察到对训练数据中重复的图像存在一定程度的记忆。可以在 [https://rom1504.github.io/clip - retrieval/](https://rom1504.github.io/clip - retrieval/) 上搜索训练数据,以帮助检测记忆的图像。
偏差
虽然图像生成模型的能力令人印象深刻,但它们也可能强化或加剧社会偏差。Stable Diffusion v1在 [LAION - 2B(en)](https://laion.ai/blog/laion - 5b/) 的子集上进行训练,该子集主要由英语描述的图像组成。使用其他语言的社区和文化的文本和图像可能没有得到充分考虑。这影响了模型的整体输出,因为白人和西方文化通常被设定为默认值。此外,模型使用非英语提示生成内容的能力明显低于使用英语提示的能力。
安全模块
该模型的预期用途是与Diffusers中的 安全检查器 一起使用。该检查器通过将模型输出与已知的硬编码NSFW概念进行比较来工作。这些概念被故意隐藏,以降低逆向工程此过滤器的可能性。具体而言,检查器在图像 生成后 将有害概念的类别概率与 CLIPTextModel
嵌入空间中的概念进行比较。这些概念与生成的图像一起传入模型,并与每个NSFW概念的手工设计权重进行比较。
训练
训练数据
模型开发者使用以下数据集进行模型训练:
- LAION - 2B (en) 及其子集(见下一节)
训练过程
Stable Diffusion v1 - 5是一个潜在扩散模型,它将自动编码器与在自动编码器的潜在空间中训练的扩散模型相结合。在训练期间:
- 图像通过编码器进行编码,将图像转换为潜在表示。自动编码器使用相对下采样因子8,将形状为H x W x 3的图像映射到形状为H/f x W/f x 4的潜在表示。
- 文本提示通过ViT - L/14文本编码器进行编码。
- 文本编码器的非池化输出通过交叉注意力输入到潜在扩散模型的UNet主干中。
- 损失是添加到潜在表示的噪声与UNet的预测之间的重建目标。
目前提供了六个Stable Diffusion检查点,它们的训练方式如下:
-
[
stable - diffusion - v1 - 1
](https://huggingface.co/CompVis/stable - diffusion - v1 - 1):在 [laion2B - en](https://huggingface.co/datasets/laion/laion2B - en) 上以分辨率256x256
训练237,000步。在 [laion - high - resolution](https://huggingface.co/datasets/laion/laion - high - resolution)(来自LAION - 5B的170M个分辨率>= 1024x1024
的示例)上以分辨率512x512
训练194,000步。 -
[
stable - diffusion - v1 - 2
](https://huggingface.co/CompVis/stable - diffusion - v1 - 2):从stable - diffusion - v1 - 1
继续训练。在 "laion - improved - aesthetics"(laion2B - en的一个子集,过滤为原始大小>= 512x512
、估计美学分数> 5.0
且估计水印概率< 0.5
的图像。水印估计来自LAION - 5B元数据,美学分数使用 [改进的美学估计器](https://github.com/christophschuhmann/improved - aesthetic - predictor) 进行估计)上以分辨率512x512
训练515,000步。 -
[
stable - diffusion - v1 - 3
](https://huggingface.co/CompVis/stable - diffusion - v1 - 3):从stable - diffusion - v1 - 2
继续训练 - 在 "laion - improved - aesthetics" 上以分辨率512x512
训练195,000步,并丢弃10%的文本条件以改进 无分类器引导采样。 -
[
stable - diffusion - v1 - 4
](https://huggingface.co/CompVis/stable - diffusion - v1 - 4) 从stable - diffusion - v1 - 2
继续训练 - 在 "laion - aesthetics v2 5+" 上以分辨率512x512
训练225,000步,并丢弃10%的文本条件以改进 无分类器引导采样。 -
[
stable - diffusion - v1 - 5
](https://huggingface.co/runwayml/stable - diffusion - v1 - 5) 从stable - diffusion - v1 - 2
继续训练 - 在 "laion - aesthetics v2 5+" 上以分辨率512x512
训练595,000步,并丢弃10%的文本条件以改进 无分类器引导采样。 -
[
stable - diffusion - inpainting
](https://huggingface.co/runwayml/stable - diffusion - inpainting) 从stable - diffusion - v1 - 5
继续训练 - 然后在 “laion - aesthetics v2 5+” 上以分辨率512x512进行440,000步的修复训练,并丢弃10%的文本条件。对于修复,UNet有5个额外的输入通道(4个用于编码的掩码图像,1个用于掩码本身),其权重在恢复非修复检查点后初始化为零。在训练期间,我们生成合成掩码,并在25%的情况下对所有内容进行掩码。 -
硬件:32 x 8 x A100 GPU
-
优化器:AdamW
-
梯度累积:2
-
批次大小:32 x 8 x 2 x 4 = 2048
-
学习率:在10,000步内预热到0.0001,然后保持不变
评估结果
使用不同的无分类器引导尺度(1.5、2.0、3.0、4.0、5.0、6.0、7.0、8.0)和50个PNDM/PLMS采样步骤进行的评估显示了检查点的相对改进:

使用50个PLMS步骤和来自COCO2017验证集的10000个随机提示进行评估,在512x512分辨率下进行评估。未针对FID分数进行优化。
环境影响
Stable Diffusion v1估计排放量
根据这些信息,我们使用 Lacoste等人(2019) 中提出的 机器学习影响计算器 估计了以下CO2排放量。利用硬件、运行时间、云服务提供商和计算区域来估计碳影响。
- 硬件类型:A100 PCIe 40GB
- 使用小时数:150000
- 云服务提供商:AWS
- 计算区域:美国东部
- 碳排放(功耗 x 时间 x 基于电网位置产生的碳):11250 kg CO2 eq.
引用
@InProceedings{Rombach_2022_CVPR,
author = {Rombach, Robin and Blattmann, Andreas and Lorenz, Dominik and Esser, Patrick and Ommer, Bj\"orn},
title = {High-Resolution Image Synthesis With Latent Diffusion Models},
booktitle = {Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR)},
month = {June},
year = {2022},
pages = {10684-10695}
}
注意:本模型卡片由Robin Rombach和Patrick Esser编写,基于 [DALL - E Mini模型卡片](https://huggingface.co/dalle - mini/dalle - mini)。
⚠️ 重要提示
此仓库是现已弃用的ruwnayml/stable - diffusion - v1 - 5的镜像,此仓库或组织与RunwayML没有任何关联。如果您是相关责任方,可以要求进行法律移除。
此模型是开放访问的,可供所有人使用,CreativeML OpenRAIL - M许可证进一步规定了权利和使用方式。CreativeML OpenRAIL许可证规定:
- 您不能使用该模型故意生成或分享非法或有害的输出或内容。
- CompVis对您生成的输出不主张任何权利,您可以自由使用它们,并对其使用负责,且使用不得违反许可证中规定的条款。
- 您可以重新分发权重,并将模型用于商业用途和/或作为服务使用。如果您这样做,请注意您必须包含与许可证中相同的使用限制,并向所有用户分享CreativeML OpenRAIL - M的副本(请完整仔细阅读许可证)。 请在此处仔细阅读完整的许可证:[https://huggingface.co/spaces/CompVis/stable - diffusion - license](https://huggingface.co/spaces/CompVis/stable - diffusion - license)

