Google Colab で SDXL 1.0を試す
「Google Colab」で「SDXL 1.0」(Stable Diffusion XL 1.0)を試したのでまとめました。
1. SDXL 1.0
「SDXL 1.0」(Stable Diffusion XL 1.0) は、「Stability AI」が開発した最新の画像生成AIです。
2. Colabでの実行
Google Colabでの実行手順は、次のとおりです。
(1) 新規のColabのノートブックを開き、メニュー「編集 → ノートブックの設定」で「GPU」を選択。
(2) パッケージのインストール。
# パッケージのインストール
!pip install diffusers
!pip install invisible_watermark transformers accelerate safetensors(4) パイプラインの準備
from diffusers import DiffusionPipeline
import torch
# パイプラインの準備
pipe = DiffusionPipeline.from_pretrained(
"stabilityai/stable-diffusion-xl-base-1.0",
torch_dtype=torch.float16,
use_safetensors=True,
variant="fp16"
)
pipe.to("cuda")GPUメモリが足りない場合は、pipe.to("cuda") を pipe.enable_model_cpu_offload() に変更することで、速度は遅くなりますがメモリ削減 (CPUオフロード) できます。
(5) 画像生成の実行。
40秒ほどで、1024x1024の画像が生成されました。
# 画像生成の実行
prompt = "cute cat ear maid"
image = pipe(prompt=prompt).images[0]
# 保存と表示
image.save("output.png")
image