Google Colab で SSD-1B を試す

「Google Colab」で「SSD-1B」を試したので、まとめました。

1. SSD-1B

「SSD-1B」(Segmind Stable Diffusion Model) は、「SDXL」(Stable Diffusion XL) を50% 小型化したバージョンで、高品質のテキストから画像への生成機能を維持しながら60% の高速化を実現します。

segmind/SSD-1B · Hugging Face

2. Colabでの実行

Colabでの実行手順は、次のとおりです。

(1) Colabのノートブックを開き、メニュー「編集 → ノートブックの設定」で「GPU」を選択。

(2) パッケージのインストール。

# パッケージのインストール
!pip install git+https://github.com/huggingface/diffusers
!pip install transformers accelerate safetensors

(3) パイプラインの準備。

from diffusers import StableDiffusionXLPipeline
import torch

# パイプラインの準備
pipe = StableDiffusionXLPipeline.from_pretrained(
    "segmind/SSD-1B", 
    torch_dtype=torch.float16, 
    use_safetensors=True, 
    variant="fp16"
)
pipe.to("cuda")

(4) 画像生成。
1024x1204の画像生成にT4で5.25秒かかりました。

%time
from transformers.pipelines.image_to_text import Image
from IPython.display import display

# プロンプトの準備
prompt = "cute cat ear maid　of japanese manga style"  # プロンプト
neg_prompt = "ugly, blurry, poor quality"  # ネガティブプロンプト

# 画像生成
image = pipe(prompt=prompt, negative_prompt=neg_prompt).images[0]

# 確認
display(image)

ChatGPT 使いこなし＆活用術