コマンドラインでStableDiffusion実行(AMD GPU/CPU対応)

やりたいこと

stable-diffusion.cppを使用してコマンドラインからStableDiffusionを実行します。

AMDのGPUでもCPUでも実行可能です。

以下のページから自分の環境にあったZipファイルをダウンロードします。

AMDのGPUで動かしたい場合はvulkanとつくものかrocmとつくものです。

（基本的にはvulkanでいいと思います。rocmはどうするGPUが限られるはずです。）

NVidiaのGPUはcudaが付くものが対象です。

avx512,avx2,avx,noavxはCPU動作です。お使いのCPUで使用できるavxのバージョンを調べてダウンロードしてください。（私は誤解していたのですがAMDのCPUもAVX使えるようです。どのバージョンが使用できるかはAIに聞いちゃうのが楽です。）

対象のダウンロードしたファイルを任意のフォルダに解凍したら準備完了です。

StableDiffusionのモデルをローカルに持っていない場合は以下のページを参考にダウンロードしてください。（モデルの保存先はどこでもいいです）

コマンドラインを起動してstable-diffusion.cppを解凍したフォルダに移動します。

以下のコマンドを実行します。（モデルパス部分は使用するmodelのパスに置き換えてください）

sd-cli -m モデルパス -p "a lovely cat" -s -1

./output.pngに猫の画像が生成されれば成功です。

オプションは以下のページにまとめられています。

よく使われる基本的なものだけ以下に記載します。

-m	モデルのパス
-p	プロンプト
-s	シード値ランダムで生成するには-1を指定する。指定しない場合、毎回同じ画像が生成されるため注意。
`-H`	画像高さ
-W	画像幅
`--vae`	VAEのパス
`--steps`	ステップ。初期値20 モデルによっては少ない数字の方がいいので注意。

以下のモデルで動作することを確認しました。

画像の生成速度は以下の通りです。（モデルのロード時間やイテレーション後の時間は含んでいません）

	生成時間（ｓ）
CPU（StableDiffusionWebui)	263
GPU（StableDiffusionWebui)	63
CPU（stable-diffusion.cpp　AVX2）	249
GPU（stable-diffusion.cpp　Vulkan）	36

参考