Stable Diffusion

「Paperspace」でStable Diffusionを動かす方法。ロースペックPCでもさくさく画像生成!

Stable Diffusionとは、あなたの入力した文字列をもとに、まるで人間の手が描いたような芸術的な画像を生成するAIです。

本記事では、一流のクラウドサービスであるPaperspaceを用いて、この驚異的なAI、Stable Diffusionを運用する方法を解説します。

Paperspaceアカウントの設定

まず初めに、Paperspaceアカウントの作成を行います。以下のリンクからPaperspaceの公式ウェブサイトにアクセスしてください。

https://www.paperspace.com/

画面右上に表示される「Sign Up」ボタンをクリックし、メールアドレスとパスワードを入力し、アカウントを作成します。

プロモーションコード欄に「BING」を入力することで、$10分の無料クレジットを受け取ることができます。

Gradientノートブックの設定

次に、Paperspaceのパワフルな機能であるGradientを用いて、ノートブックを作成します。Gradientは無料でGPUを利用可能なクラウドサービスです。

ウェブサイトの左側にあるメニューから、「Gradient」アイコンをクリックし、「Notebooks」タブを選択します。

「Create Notebook」ボタンをクリックし、次の設定を行います:

  • Notebook name: お好みの名前を入力します。「Stable Diffusion」などが適しています。
  • Container: 「Python 3 + PyTorch 1.9 + CUDA 11.1」を選択します。
  • Machine type: 「Free-GPU」を選択します。
  • Workspace: 「GitHub」を選択し、「Repository URL」に「https://github.com/openai/stable-diffusion.git」と入力します。

設定が完了したら、「Create Notebook」ボタンをクリックします。ノートブックの作成には数分かかることがあります。

Stable Diffusion WebUIのインストールと画像生成

最後に、Stable Diffusion WebUIをインストールし、画像生成を行います。

Stable Diffusion WebUIは、ブラウザから直感的に操作できる画像生成ツールです。

ノートブックの最初のセルに以下のコードを貼り付けて実行します。

!pip install flask-ngrok
!python webui.py --model_path stable_diffusion_256x256/model.ckpt-1000000.index

コードの実行後、以下のようなメッセージが表示されます。

* Running on http://127.0.0.1:5000/ (Press CTRL+C to quit)
* Running on http://xxxx-xxxx-xxxx-xxxx.ngrok.io
* Traffic stats available on http://127.0.0.1:4040

表示されたhttp://xxxx-xxxx-xxxx-xxxx.ngrok.ioというリンクをクリックすると、Stable Diffusion WebUIが起動します。

WebUIで以下の操作を行うことができます

  • 「Text」欄に、画像生成を希望する文字列を入力します。例えば、「動物」と入力してみましょう。
  • 「Generate」ボタンをクリックすると、入力した文字列に基づいて画像が生成されます。生成時間は数秒から数十秒となることがあります。
  • 生成された画像は、「Save」ボタンをクリックすることでローカル環境に保存することができます。
  • 「Text」欄の内容を変更し、「Generate」ボタンをクリックすると、新たな画像が生成されます。同じ文字列を用いても、毎回異なる画像が生成されるという点に留意してください。

以上で、Paperspaceを活用したStable Diffusionの運用方法の紹介を終えます。この記事が皆様の役に立てれば幸いです。

Paperspaceを通じて、あなた自身の想像力を絵画に変換する楽しい体験をぜひともお楽しみください。