感谢与Stability AI以及Runway的合作,再加上我们先前的一些工作使我们能够做出Stable Diffusion:
High-Resolution Image Synthesis with Latent Diffusion Models
Robin Rombach*,
Andreas Blattmann*,
Dominik Lorenz,
Patrick Esser,
Björn Ommer
CVPR '22 Oral |
GitHub | arXiv | Project page
Stable Diffusion是一个隐式的文生图扩散模型。感谢Stability AI提供了充足的算力支持,以及LAION提供的数据支撑,使得我们能够在LAION-5B数据集上采用
512x512
分辨率训练一个隐式扩散模型。和谷歌的Imagen类似,该模型使用了一个冻结权重的CLIP ViT-L/14文本编码器,利用该编码器得到控制模型的文本提示。该模型包含了一个860M UNet以及一个123M text encoder,相对而言比较轻量使用一个拥有10G显存的GPU就能运行。
可通过如下指令创建一个名为ldm
的conda环境:
conda env create -f environment.yaml
conda activate ldm
创建好上述环境后,你也可以通过如下指令更新latent diffusion环境:
conda install pytorch torchvision -c pytorch
pip install transformers==4.19.2 diffusers invisible-watermark
pip install -e .
Stable Diffusion v1采用的是扩散模型架构,其主要由三部分组成: (1) 下采样倍率为8的自编码器 (2) 参数规模860M的UNet (3) CLIP ViT-L/14文本编码器。Stable Diffusion v1是先在256x256
分辨率的图像上进行预训练,然后在512x512
分辨率的图像上进行finetune。
注意: Stable Diffusion v1是一个通用的文生图扩散模型,因此生成的结果会带上来自训练数据中的偏见和错误概念。有关训练过程,训练数据,模型用途等相关细节可参考model card.
权重文件可通过the CompVis organization at Hugging Face下载,并遵守The CreativeML OpenRAIL M license条款。虽然该条款允许商业使用,但我们不建议在没有额外安全机制以及考虑的情况下使用开源权重。因为该模型权重存在已知的限制和偏见,并且关于通用文生图模型的安全、道德部署研究正在进行中。模型权重是研究探索过程中的产物,理应如此看待。
The CreativeML OpenRAIL M license is an Open RAIL M license, adapted from the work that BigScience and the RAIL Initiative are jointly carrying in the area of responsible AI licensing. See also the article about the BLOOM Open RAIL license on which our license is based.
当前,我们提供了如下权重:
sd-v1-1.ckpt
: 先在laion2B-en数据集上采用256x256
分辨率训练迭代了237k steps,接着在laion-high-resolution数据集上采用512x512
分辨率训练迭代了194k steps.sd-v1-2.ckpt
: 基于sd-v1-1.ckpt
权重在laion-aesthetics v2 5+数据集上采用512x512
分辨率又训练迭代了515k steps(laion-aesthetics v2 5+
是laion2B-en
的子集,该子集美学评估分数大于5.0
,过滤掉了原始分辨率大于等于512x512
的图像,水印评估概率小于0.5
,其中水印评估概率来自LAION-5B metadata,美学评估分数使用LAION-Aesthetics Predictor V2计算得到)sd-v1-3.ckpt
: 基于sd-v1-2.ckpt
权重在"laion-aesthetics v2 5+"数据集上采用512x512
分辨率接着训练迭代了195k step,并且训练过程中会丢弃10%的文本条件信息以提升classifier-free guidance sampling.sd-v1-4.ckpt
: 基于sd-v1-2.ckpt
权重在"laion-aesthetics v2 5+"数据集上采用512x512
分辨率接着训练迭代了225k step,并且训练过程中会丢弃10%的文本条件信息以提升classifier-free guidance sampling.
下图展示了不同的训练权重采用不同的classifier-free guidance scales (1.5, 2.0, 3.0, 4.0,
5.0, 6.0, 7.0, 8.0)以及固定50步的PLMS sampling结果:
Stable Diffusion是由CLIP ViT-L/14文本编码器生成的text embeddings控制的隐式扩散模型。 我们提供了一个可供参考的采样脚本, 但同时这也有一个已存在的diffusers integration库, 我们期待这个库对应的开源社区能够更加活跃。
我们提供了一个参考脚本,其中包括了
- 一个安全检查模块, 以避免生成一些不好的内容。
- 在输出图像中加入了隐藏水印 可帮助查看者鉴别该图像是否由机器生成。
在获取到stable-diffusion-v1-*-original
权重后, 通过软连接的形式链接它。
mkdir -p models/ldm/stable-diffusion-v1/
ln -s <path/to/model.ckpt> models/ldm/stable-diffusion-v1/model.ckpt
接着使用如下指令进行采样:
python scripts/txt2img.py --prompt "a photograph of an astronaut riding a horse" --plms
这里默认使用的是Katherine Crowson's implementation的PLMS 采样器,并且将guidance scale默认设置为--scale 7.5
, 并且生成的图像分辨率是512x512
(保持和训练时一致)且采样步数为50步。完整的参数可通过(python scripts/txt2img.py --help
)查看。
usage: txt2img.py [-h] [--prompt [PROMPT]] [--outdir [OUTDIR]] [--skip_grid] [--skip_save] [--ddim_steps DDIM_STEPS] [--plms] [--laion400m] [--fixed_code] [--ddim_eta DDIM_ETA]
[--n_iter N_ITER] [--H H] [--W W] [--C C] [--f F] [--n_samples N_SAMPLES] [--n_rows N_ROWS] [--scale SCALE] [--from-file FROM_FILE] [--config CONFIG] [--ckpt CKPT]
[--seed SEED] [--precision {full,autocast}]
optional arguments:
-h, --help 展示帮助文档信息
--prompt [PROMPT] 输入想要生成图像的描述语句
--outdir [OUTDIR] 指定生成结果保存目录
--skip_grid 跳过将所有生成图片保存成一张大图,当需要评估大批量生成图像时有用
--skip_save 跳过单独保存每张生成图像(在测速时可跳过)
--ddim_steps DDIM_STEPS 扩散模型的迭代次数
--plms 使用plms sampling
--laion400m uses the LAION400M model
--fixed_code if enabled, uses the same starting code across samples
--ddim_eta DDIM_ETA ddim eta (eta=0.0 corresponds to deterministic sampling
--n_iter N_ITER sample this often
--H H 生成图像的高度
--W W 生成图像的宽度
--C C 隐空间的通道数
--f F 图像到隐空间的下采样倍率
--n_samples N_SAMPLES
推理时采用的batch_size可以理解同时生成图像的数目, 如果显存少可设置小点
--n_rows N_ROWS rows in the grid (default: n_samples)
--scale SCALE unconditional guidance scale: eps = eps(x, empty) + scale * (eps(x, cond) - eps(x, empty))
--from-file FROM_FILE
if specified, load prompts from this file
--config CONFIG path to config which constructs model
--ckpt CKPT path to checkpoint of model
--seed SEED the seed (for reproducible sampling)
--precision {full,autocast}
evaluate at this precision
注意:在configs/stable-diffusion/v1-inference.yaml
配置中已将use_ema
设置为False
,这样你可以下载不带EMA
的权重(文件大小会小很多),如果你不知道如何区分有没有带EMA
的权重可以直接下载带有full
关键字的权重例如sd-v1-4-full-ema.ckpt
,这样在载入权重时会自动根据use_ema
参数载入对应权重。
另一种更简单的方式是直接使用diffusers库,具体环境配置可参考diffusers library:
# make sure you're logged in with `huggingface-cli login`
from torch import autocast
from diffusers import StableDiffusionPipeline
pipe = StableDiffusionPipeline.from_pretrained(
"CompVis/stable-diffusion-v1-4",
use_auth_token=True
).to("cuda")
prompt = "a photo of an astronaut riding a horse on mars"
with autocast("cuda"):
image = pipe(prompt)["sample"][0]
image.save("astronaut_rides_horse.png")
通过SDEdit提出的扩散去噪机制,该模型适用于多种任务,例如基于本文引导的图生图转换,图像超分等。与txt2img脚本类似,我们提供了一个利用Stable Diffusion对图像进行转换生成的脚本。
在下面示例中,利用一张手绘草图转换生成出一张充满细节的艺术品。
python scripts/img2img.py --prompt "A fantasy landscape, trending on artstation" --init-img <path-to-img.jpg> --strength 0.8
其中strength是一个0-1之间的参数,该参数用于控制添加到原图上噪声的比例。数值越接近1生成的多样性就越高但语义不一致的现象越明显。下面是一些示例。
Input
Outputs
-
我们扩散模型代码主要是建立在OpenAI's ADM codebase以及https://github.com/lucidrains/denoising-diffusion-pytorch。开源万岁!
-
transformer encoder的实现来自lucidrains的x-transformers。
@misc{rombach2021highresolution,
title={High-Resolution Image Synthesis with Latent Diffusion Models},
author={Robin Rombach and Andreas Blattmann and Dominik Lorenz and Patrick Esser and Björn Ommer},
year={2021},
eprint={2112.10752},
archivePrefix={arXiv},
primaryClass={cs.CV}
}