使用

from diffusers import StableDiffusionPipeline

pipe = StableDiffusionPipeline.from_pretrained("lambdalabs/miniSD-diffusers")
pipe = pipe.to("cuda")

prompt = "a photograph of an wrinkly old man laughing"
image = pipe(prompt, width=256, height=256).images[0]  
image.save('test.jpg')

培训细节

从稳定扩散1.4的检查点进行微调，如下：

• 22,000步仅微调unet的注意力层，学习率=1e-5，批量大小=256

• 66,000步训练完整的unet，学习率=5e-5，批量大小=552

• 由 Lambda GPU Cloud 提供的GPU

• 训练于 LAION Improved Aesthetics 6plus

• 使用 https://github.com/justinpinkney/stable-diffusion 进行训练，原始 checkpoint available here

许可证

此模型是开放访问的，对所有人可用，具有CreativeML OpenRAIL-M许可证来进一步指定权利和使用方式。CreativeML OpenRAIL许可证规定：

• 您不得使用该模型故意产生或共享非法或有害的输出或内容
• 作者对您生成的输出不享有任何权利，您可自由使用它们，并对使用它们负责，不得违反许可证中的规定
• 您可以重新分发权重并将该模型用于商业和/或作为服务。如果您这样做，请注意您必须包括与许可证中相同的使用限制，并向您所有用户共享CreativeML OpenRAIL-M的副本（请完整而仔细地阅读许可证）请阅读完整的许可证