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