話題の Stable DiffusionをColabで作ってみた！

今話題のAI画像生成の簡易的なソースを作りましたので、試してみたい方は是非使ってみてください！

実行環境

Google colab
Python 3.7.13
diffusers 0.2.4

前提条件

Googleのアカウントがある
huggingfaceのアカウントがある

環境構築手順

1. ライセンスの確認
   モデルカードにアクセスしてライセンスを確認します。
   https://huggingface.co/CompVis/stable-diffusion-v1-4
   

2. アクセストークン発行
   https://huggingface.co/settings/tokens
   

3. Colabハードウェアアクセラレータの変更
   通常NoneになっているハードウェアアクセラレータをGPUに変更
   Colab上部タブ「編集」押下
   ↓
   ノートブックの編集を押下
   ↓
   ハードウェアアクセラレータのプルダウンでGPUを選択

Colabコマンド

!pip install diffusers==0.2.4 transformers scipy ftfy

import torch
from torch import autocast
from diffusers import StableDiffusionPipeline

model_id = "CompVis/stable-diffusion-v1-4"
device = "cuda"
YOUR_TOKEN="<HugginFace Hubのトークン>"

pipe = StableDiffusionPipeline.from_pretrained(model_id, use_auth_token=YOUR_TOKEN)

# お題をpromptに記入
prompt = "日本 城 合戦"
image = pipe(prompt)["sample"][0]

# 保存名を記載
image.save(f"japanese_castle.png")

実行すると下記の写真のように処理が始まり、完了するまで放置する。

上記のキーワードでできた画像がこちら

お城というかお寺ができましたね！