🚀 SD v2.1-base with Zero Terminal SNR (LAION Aesthetic 5+)

这是论文 Common Diffusion Noise Schedules and Sample Steps are Flawed 的官方模型。该模型使用零终端信噪比（Zero Terminal SNR）调度在 LAION aesthetic 5+ 数据上进行训练，是在 stabilityai/stable-diffusion-2-1-base 基础上微调得到的。此模型仅用于研究展示，不用于生产环境。

🚀 快速开始

💻 使用示例

基础用法

from diffusers import StableDiffusionPipeline
prompt = "A young girl smiling"
pipe = StableDiffusionPipeline.from_pretrained("ByteDance/sd2.1-base-zsnr-laionaes5").to("cuda")
pipe(prompt, guidance_scale=7.5, guidance_rescale=0.7).images[0].save("out.jpg")

📚 详细文档

相关模型

• bytedance/sd2.1-base-zsnr-laionaes5
• bytedance/sd2.1-base-zsnr-laionaes6
• bytedance/sd2.1-base-zsnr-laionaes6-perceptual

引用方式

@misc{lin2023common,
      title={Common Diffusion Noise Schedules and Sample Steps are Flawed}, 
      author={Shanchuan Lin and Bingchen Liu and Jiashi Li and Xiao Yang},
      year={2023},
      eprint={2305.08891},
      archivePrefix={arXiv},
      primaryClass={cs.CV}
}

📄 许可证

该模型采用 OpenRAIL++ 许可证。