Google Colab で SDXL Turbo を試す

「Google Colab」で「SDXL Turbo」を試したので、まとめました。

1. SDXL Turbo

「SDXL Turbo」は、「Stability AI」が開発した画像生成AIです。新しい蒸留技術によって、これまでにない品質のシングルステップ画像生成を可能にしました。必要なステップ数は50から1に削減されました。

リアルタイムテキスト画像生成モデル、SDXL Turboのご紹介 — Stability AI Japan

2. Colabでの実行

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

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

# パッケージのインストール
!pip install diffusers transformers accelerate --upgrade

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

from diffusers import AutoPipelineForText2Image
import torch

# パイプラインの準備
pipe = AutoPipelineForText2Image.from_pretrained(
    "stabilityai/sdxl-turbo", 
    torch_dtype=torch.float16, 
    variant="fp16"
).to("cuda")

(3) 画像生成。
697msで生成できました。

%%time

# プロンプトの準備
prompt = "cute cat ear maid"

# 画像生成
image = pipe(
    prompt=prompt, 
    num_inference_steps=1, 
    guidance_scale=0.0
).images[0]

# 確認
image

ChatGPT 使いこなし＆活用術