2022年09月02日 13時00分ソフトウェア

画像生成AI「Stable Diffusion」をM1搭載Macのローカル上で実行する方法

非常に高精度な画像を生成できるとして大きな話題を呼ぶ画像生成AI「Stable Diffusion」は、オープンソースであるためユーザーがローカル環境で実行することもできますが、記事作成時点ではNVIDIA製GPUの使用が推奨されています。そのため、「Stable Diffusionが気になっているけどMacユーザーだからローカルで実行できない」という人もいるはず。そんなMacユーザーに向けて、「Stable DiffusionをM1・M2チップ搭載Macのローカル上で実行する方法」についてオープンソースの機械学習モデルをクラウドのAPIで実行するウェブサービス・ Replicateが解説しています。

Run Stable Diffusion on your M1 Mac’s GPU - Replicate – Replicate
https://replicate.com/blog/run-stable-diffusion-on-m1-mac

ReplicateはオープンソースのAIをクラウドで実行できるAPIを開発しており、Stable Diffusionをクラウドで実行するAPIも提供しています。また、ReplicateはStable DiffusionをM1・M2チップを搭載したMacのローカル上で実行する方法についても、公式ブログで説明しています。

前提条件として、利用するデバイスはM1またはM2チップを搭載したMacで、OSはmacOS 12.3以降のバージョンであることが必要です。また、8GBのRAMでも動作するものの非常に遅いため、可能であれば16GB以上のRAMを搭載していることが望ましいとのこと。

まず、Stable Diffusionを実行するにはPython 3.10以降のバージョンが必要なため、インストールしていない場合はパッケージ管理システムのHomebrewから最新バージョンのPythonをインストールします。

brew update
brew install python

最新バージョンのPythonをインストールしたら、Stable Diffusionのフォークをクローンします。

git clone -b apple-silicon-mps-support https://github.com/bfirsh/stable-diffusion.git
cd stable-diffusion
mkdir -p models/ldm/stable-diffusion-v1/

続いて、依存関係をインストールするためにVirtualenvをセットアップして……

python3 -m pip install virtualenv
python3 -m virtualenv venv

アクティブにします。なお、Stable Diffusionを実行するたびにこのコマンドを再実行する必要があるとのこと。

source venv/bin/activate

次にPythonパッケージのインストール・管理システムであるpipから依存関係をインストールします。

pip install -r requirements.txt

このようなエラーが表示される場合……

Failed building wheel for onnx

以下のパッケージをインストールする必要があるかもしれないとのこと。

brew install Cmake protobuf rust

そして自然言語処理のオープンソースコミュニティ・Hugging Faceのリポジトリに移動してライセンスを読み、「Access repository」をクリック。

「sd-v1-4.ckpt(～4GB)」をダウンロードし、上記で作成したディレクトリ「models/ldm/stable-diffusion-v1/」に「model.ckpt」という名前で保存します。

models/ldm/stable-diffusion-v1/model.ckpt

これでセットアップは完了です。以下のようにコードを実行し、「a red juicy apple floating in outer space, like a planet(赤くてジューシーなリンゴが、まるで惑星のように宇宙空間に浮かんでいる)」という文字列を入力すると……

python scripts/txt2img.py \
  --prompt "a red juicy apple floating in outer space, like a planet" \
  --n_samples 1 --n_iter 1 --plms

「outputs/txt2img-samples/」に以下のような画像が生成されるとのことです。

Replicateは、この方法は自分たちが独自に編み出したものではなく、すべての功績はGitHubで公開されているStable Diffusionのフォークに貢献したり、GitHubのスレッドに参加したりした開発者にあると主張。以前の作業に付け加えたのは、pipを使用して依存関係をインストールすることでセットアップを簡略化した点くらいだとして、「私たちは彼らの偉大な働きのメッセンジャーに過ぎません」と述べています。

なお、Stable DiffusionをWindows環境にインストールしてGUIで操作できる「NMKD Stable Diffusion GUI」や、Intel製GPUで実行できる「stable_diffusion.openvino」なども登場しています。

画像生成AI「Stable Diffusion」をWindows環境にボタン1つでインストール可能＆GUIで操作できる「NMKD Stable Diffusion GUI」がついに登場 - GIGAZINE