Small Sd

🚀 テキストから画像への蒸留

このパイプラインは、SG161222/Realistic_Vision_V4.0 を recastai/LAION-art-EN-improved-captions データセットのサブセットで蒸留したものです。以下は、small - sdモデルを使用して微調整されたパイプラインで生成されたサンプル画像です。

このパイプラインは この論文 に基づいています。トレーニングコードは こちら で見つけることができます。

🚀 クイックスタート

✨ 主な機能

このパイプラインは、テキスト入力から画像を生成することができます。SG161222/Realistic_Vision_V4.0 をベースに、特定のデータセットで蒸留されています。

💻 使用例

基本的な使用法

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 ライセンスの下で提供されています。

情報テーブル