Stable DiffusionをGoogle Colaboratoryで簡単に立ち上げて使う方法を紹介します。
結論から言うと以下の通りに作業を進めていけばOKです。
※記事の途中でエラーが出た場合の対処法を記載しています。
- Google Colaboratoryの有料プランと契約
- 新規ノートブックの作成
- GPUを有効化する
- 必要なライブラリをインストール
- Stable Diffusionをインストール
- モデルのダウンロード
- VAEのダウンロード
- 実行
- .liveのページにアクセス
- プロンプトを入力して画像を生成
では順番に説明していきます。
Google Colaboratoryの有料プランと契約
まずは、以下記事を参考にColabの有料プランと契約してください。
新規ノートブックの作成
ファイルの「ノートブックを新規作成」をクリック。
GPUを有効化する
ランタイムより「ランタイムのタイプを変更」を開きます。
最初はコンピューティングユニットの消費量が少ないT4 GPUを選択し、保存します。
以下の赤枠部が選択したアクセラレータT4 GPUに切り替わります。
必要なライブラリをインストール
中央の「コーディングを開始するか、AIで生成します。」と書かれた箇所をクリックし、以下のコードを入力してShift+Enterを実行してください。
!pip install lmdb
!pip install torch==2.0.1+cu118 torchvision==0.15.2+cu118 torchaudio==2.0.2 torchtext==0.15.2+cpu torchdata==0.6.1 --index-url https://download.pytorch.org/whl/cu118※ここで以下のようなxformersのエラーが発生した場合は、githubの記事を参考に解決してください。
参考：https://github.com/TheLastBen/fast-stable-diffusion/issues/2615
上のエラーに限らず、colab使用中にエラーメッセージが出たら、githubのissuesに内容を打ち込んで検索すると解決策が見つかることがあるので参考にしてください。
Stable Diffusionをインストール
以下コードを実行し、Stable Diffusionをインストールしてください。
!git clone https://github.com/AUTOMATIC1111/stable-diffusion-webui次に、以下コマンドにより、Stable Diffusionがインストールされたフォルダへ移動します。
%cd /content/stable-diffusion-webuiモデルのダウンロード
以下コマンドでモデルをダウンロードします。
今回はアニメ系美少女を生成できるAnyLoraというモデルをダウンロードします。
参考：https://civitai.com/models/23900?modelVersionId=95489
!wget https://civitai.com/api/download/models/95489 -O /content/stable-diffusion-webui/models/Stable-diffusion/AnyLora.safetensorsVAEのダウンロード
続いてVAEもダウンロードしておきましょう。
!wget https://huggingface.co/stabilityai/sd-vae-ft-mse-original/resolve/main/vae-ft-mse-840000-ema-pruned.safetensors -O /content/stable-diffusion-webui/models/VAE/vae-ft-mse-840000-ema-pruned.safetensors実行
最後に、以下コマンドで実行します。
!python launch.py --share --xformers --enable-insecure-extension-access.liveのページにアクセス
5〜10分くらいでURLの最後に.liveがついたページが作成されるので、こちらを開きましょう。
プロンプトを入力して画像を生成
あとは、プロンプトやネガティブプロンプトを入力して、生成ボタンをクリックすればひとまずアニメ系画像が作成されるはずです。
まず以下画像の赤矢印箇所で先ほどダウンロードしたモデルのAnyLora.safetensorsが選択されていることを確認してください。
※コマンド入力にミスがあるなどしてダウンロードに失敗している場合はやり直してください。
次にプロンプトとネガティブプロンプトを入力し、右上オレンジのGenerateボタンをクリックすればAI画像が生成されます。
プロンプトがよくわからない方は、「girl」と入力してGenerateしてみてください。
先ほどのAnyLoraのサイトに掲載されていたアニメ少女に似た画像が生成されれば成功です。
以上です。
お疲れ様でした。
コメント