Stable DiffusionをGoogle Colabで使いたい！

こんにちは！サキナオコです^o^
日差しはもうすっかり春ですね。

はい、で、もう頭の中がAIでいっぱいの私です。
呪文の修行も早くしたいのですが、その前にStable DiffusionをGoogle Colabで使ってみたいのです！

天下のChatGPT様に聞いてやってみましたが、エラーが出て何度やってもどうしてもダメでした。

ウェブ情報で調べてやってみたらStable DiffusionでなくImaginAIryというモデルだったので、再度挑戦。いや、いいんです、ImaginAIryでもね！でも目標はStable Diffutionなんです！

★GoogleColabにログインしたらメニュー＞編集＞ノートブックの設定＞ハードウェア アクセラレータ>GPUにしておく　これ大事！

情報ではHugging Faceというものを使ってできるようなので、
いろいろやり方を調べてやってみました。

★Hugging Faceでやること

• まずは黄色のお化けみたいなキャラのHugging Faceに登録。
• メールで認証後ログイン
• 右上ユーザーアイコン〜
  Setting>AccessTokenでNew tokenを readとwriteそれぞれ作成(これって必要なのかな？わからない。結局使ってない)←いらないらしい(2023.3.13)
• 左上serchウィンドウでStableと入れるとモデルがずらりと出る
• バージョンを適当にクリックして、実際にテストで画像生成してみる
• よければGoogle Colabにもどり以下を入力

!pip install diffusers[torch]==0.8.1 transformers

• Hugging Faceにもどり以下のUse in Diffusersをクリックすると

コードが出るので上のをコピー。

• GoogleColabに出たコードをペーストしてあと追加してウェブにあったコードを書く

from diffusers import DiffusionPipeline
import torch
pipe = DiffusionPipeline.from_pretrained("runwayml/stable-diffusion-v1-5",
torch_dtype=torch.float16, revision="fp16").to("cuda")

私が選んだバージョンのです↑pipe=というところはpipelineとなっていたのをpipeに変更

• GoogleDriveへ保存するなら、Colabのウィンドウ左、縦メニューの下のフォルダクリック、出てきたドライブフォルダマウントクリック、フォルダ作りたかったら作る
• 以下のコードでドライブに保存される

import os
os.chdir('/content/drive/MyDrive/aiimg')

• 次のコードのprompt= の後に呪文を書けば画像生成！

prompt = "a cup of tea on the desk"
image = pipe(prompt).images[0]  
image.save("op3.png")

• 15秒で生成できた！ドライブに保存されてます！うれし！

初めコードが足りなかったみたいで
（torch_dtype=torch.float16, revision=”fp16″).to(“cuda”)のところ）

小一時間かかって生成(泣)してましたが、
あらためてコード調べてやってみたら10秒位で生成できました！^o^

えーっとまたいろいろやってみます。これからちょっと呪文の練習！

追記：2023年4月からGoogle colab無料版でstable diffusionを使うと警告が出て利用ができなくなったようです。Pro版だと大丈夫だとのこと。