Small Sd

🚀 文本到圖像蒸餾模型

本項目是一個文本到圖像的蒸餾模型，它基於特定數據集對基礎模型進行蒸餾微調，能夠根據文本描述生成圖像。該模型的訓練代碼和相關研究有公開的參考資源，方便進一步研究和使用。

🚀 快速開始

模型簡介

此管道是在 recastai/LAION-art-EN-improved-captions 數據集的一個子集上，從 SG161222/Realistic_Vision_V4.0 進行蒸餾得到的。以下是使用微調後的管道和 small-sd 模型生成的一些示例圖像。

本管道基於這篇論文。訓練代碼可在此處找到。

管道使用方法

你可以按照以下方式使用該管道：

from diffusers import DiffusionPipeline
import torch

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

✨ 主要特性

• 基於特定數據集對基礎模型進行蒸餾微調，能夠根據文本描述生成高質量的圖像。
• 提供了公開的訓練代碼和相關研究論文，方便進一步研究和使用。

💻 使用示例

基礎用法

from diffusers import DiffusionPipeline
import torch

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

🔧 技術細節

訓練信息

以下是訓練期間使用的關鍵超參數：

• 步數：95000
• 學習率：1e - 4
• 批量大小：32
• 梯度累積步數：4
• 圖像分辨率：512
• 混合精度：fp16

📄 許可證

本項目採用 creativeml-openrail-m 許可證。