はじめに
2022/8に画像生成AI「Stable Diffusion」が登場した。
こちら、英Stability AIが発表した画像生成AI。すごい！
紹介されているサイトでも画像は生成できるので十分楽しめるそうなんだけれども
APIも公開されてるようで、Google Colaboratoryを使って実行することで
安ＰＣでもこのAIの恩恵をあずかれるってことでやってみようと思う。
まずはアカウント登録
を参考にやってみる。まずは、Hugging Faceのサイトに行ってアカウントを作成。
画面右上の「Sign UP」クリック
メアドとログインに使用するパスワードを入力して Next
プロファイル情報を入力
登録完了。
https://huggingface.co/welcome
こんな感じでどうやって使うのか紹介してくれた
gmailにメアド確認メールが届いていたのでリンクをクリック。
リンク先で↓↓の画面がでて、無事に承認されたことが確認できた。
アクセストークンを入手する
画面右上のアイコン画像をクリック＞Settingクリック
画面左側より Access Tokens クリック
New Token クリック
Nameは適当な名前を入れて、、、
Roleはread
で、 Gnenerate a token
無事アクセストークンの生成に成功 Show をクリックしたら内容が読める
この内容は後で使うのでコピーしておく
ColaboratoryでStable Diffusionをセットアップする
Google ドライブより、Google Colaboratoryを起動する
ノートブックの設定 クリック
GPUを選択して 保存
pipで必要なライブラリをインストール
ここからはColaboratoryにプログラムを入力していく。
pip install diffusers==0.2.4 transformers scipy ftfy
しばし待つ…
１分くらいで完了。
Stable Diffusionを呼び出す
さっき拾ってきたトークンを変数に代入
YOUR_TOKEN=”Hugging Faceで発行したトークン”
以下のコードを実行する
from diffusers import StableDiffusionPipeline
pipe = StableDiffusionPipeline.from_pretrained("CompVis/stable-diffusion-v1-4", use_auth_token=YOUR_TOKEN)
２分くらいかかったけど成功した
最後に↓↓のコードを実行してセットアップ完了
pipe.to("cuda")
Stable Diffusionで画像を生成する
以下のコードを実行
prompt = "cute cat paly with ball"
image = pipe(prompt)["sample"][0]
image.save(f"cat.png")
ファイル一覧を見ると、cat.pngが生成されてることがわかる。
cat.pngをダブルクリックすると、画面右側に画像が表示された。成功！
おわりに
オープンソースで画像AIが利用できるってすごい時代が始まったな。。。って素直に思いました。
しばらくはStableDiffusionで遊んでみようと思う。