見出し画像

Google Colab で SDXL 1.0を試す

npaka

「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



この記事が気に入ったらサポートをしてみませんか？