Tiny Sd

🚀 文本到图像蒸馏

此管道是从SG161222/Realistic_Vision_V4.0在recastai/LAION - art - EN - improved - captions数据集的一个子集上进行蒸馏得到的。它能够将文本描述转化为图像，为图像生成领域提供了高效的解决方案。

🚀 快速开始

此管道基于论文实现。训练代码可在此处找到。以下是使用该管道的示例，展示了如何根据文本提示生成图像并保存。

💻 使用示例

基础用法

from diffusers import DiffusionPipeline
import torch

pipeline = DiffusionPipeline.from_pretrained("segmind/tiny-sd", torch_dtype=torch.float16)
prompt = "Portrait of a pretty girl"
image = pipeline(prompt).images[0]
image.save("my_image.png")

📚 详细文档

训练信息

以下是训练过程中使用的关键超参数：

• 步数：125000
• 学习率：1e - 4
• 批量大小：32
• 梯度累积步数：4
• 图像分辨率：512
• 混合精度：fp16

速度比较

我们观察到，蒸馏模型比基础SD1.5模型快达80%。以下是在A100 80GB上的比较结果。

此处是用于对速度进行基准测试的代码。

示例图像

以下是使用tiny - sd模型生成的一些示例图像。

📄 许可证

本项目采用creativeml - openrail - m许可证。