[Stable Diffusion] 学習モデルの変更方法 (WSL+Ubuntu環境)

Stable Diffusion web UIは本当に使い勝手が良いですね。
今回は学習モデルを変更する方法を紹介します。

◆ Stable Diffusion web UIのインストール方法について

◆ [追記: 2023/03/30] 汎用的に使える学習モデルの追加方法について

学習モデルの追加と変更

学習モデルの追加方法

追加方法はシンプルです。殆どの場合で以下のディレクトリにckpt(学習モデル)とyaml(設定ファイル)を同名で配置します。

stable-diffusion-webui/models/Stable-diffusion

以下はStableDiffusionの2.0モデルを配置する場合の例です。WSLの機能を使ってWindowsから手動で配置してもいいです。

cd ~/stable-diffusion-webui
wget https://huggingface.co/stabilityai/stable-diffusion-2/resolve/main/768-v-ema.ckpt -O ./models/Stable-diffusion/768-v-ema.ckpt
wget https://raw.githubusercontent.com/Stability-AI/stablediffusion/main/configs/stable-diffusion/v2-inference-v.yaml -O ./models/Stable-diffusion/768-v-ema.yaml

wgetは単にダウンロードするコマンドです。最終的に以下の2つのファイルが配置できれば問題ありません。

stable-diffusion-webui/models/Stable-diffusion/768-v-ema.ckpt
stable-diffusion-webui/models/Stable-diffusion/768-v-ema.yaml

学習モデルの変更方法

学習モデルはUIを使って変更します。とりあえず、Ubuntu経由でStable Diffusion web UIを起動しましょう。
起動したらUIの左上の方に存在するコンボボックス(画像の赤四角)から変更したい対象を選んで、すぐ隣の青いボタン(画像の赤丸)を押します。

学習モデルの切り替え時はUbuntuのコマンドラインを見ておきましょう。と言うのも、この時にメモリ不足でプロセスが落ちることが結構あります。
たぶんWSL経由で動作させてる影響でしょう。RAMが32GBあっても落ちます。こうなるとWindows側から次のコマンドでWSL自体を再起動しないと駄目です。

wsl.exe --shutdown

いろんな学習モデル

管理人

残りは適当に学習モデルを紹介します。

学習モデルはHuggingFaceってサイトで探してきます。この時にyamlが無い場合は標準モデルのyamlで代用します。

https://huggingface.co/models?sort=likes

とか

https://huggingface.co/models?sort=downloads

で探せます。たぶん、機械学習のGitHub的なサイトです。

なお、AI自動画像生成は学習モデルに著作権画像が使用されてる等の問題を抱えてます。
個人でAIすげぇって遊ぶには問題ありませんが、販売とかは普通にNGだと思います。

Waifu Diffusion

管理人

殆どの人が使いたいであろう二次元美少女用のモデルです。

https://huggingface.co/hakurei/waifu-diffusion-v1-4

cd ~/stable-diffusion-webui
wget https://huggingface.co/hakurei/waifu-diffusion-v1-4/resolve/main/wd-1-4-anime_e2.ckpt -O ./models/Stable-diffusion/wd-1-4-anime_e2.ckpt
wget https://huggingface.co/hakurei/waifu-diffusion-v1-4/resolve/main/wd-1-4-anime_e1.yaml -O ./models/Stable-diffusion/wd-1-4-anime_e2.yaml

Counterfeit

管理人

これも二次元美少女用のモデルです。かわいい。

https://huggingface.co/gsdf/Counterfeit-V2.0

cd ~/stable-diffusion-webui
wget https://huggingface.co/gsdf/Counterfeit-V2.0/resolve/main/Counterfeit-V2.0.ckpt -O ./models/Stable-diffusion/Counterfeit-V2.0.ckpt
wget https://huggingface.co/gsdf/Counterfeit-V2.0/resolve/main/v1-inference.yaml -O ./models/Stable-diffusion/Counterfeit-V2.0.yaml

Ghibli Diffusion

管理人

ジブリ風の画像を作るモデルです。販売とかしたら普通にアウト。

https://huggingface.co/nitrosocke/Ghibli-Diffusion

cd ~/stable-diffusion-webui
wget https://huggingface.co/nitrosocke/Ghibli-Diffusion/resolve/main/ghibli-diffusion-v1.ckpt -O ./models/Stable-diffusion/ghibli-diffusion-v1.ckpt
wget https://huggingface.co/runwayml/stable-diffusion-v1-5/resolve/main/v1-inference.yaml -O ./models/Stable-diffusion/ghibli-diffusion-v1.yaml

Mo Di Diffusion

管理人

ディズニーっぽいキャラを作るモデルです。遊びの範囲を超えると夢の国から使者が来て消されます。

りさ

絶対やばいやつだ...

https://huggingface.co/nitrosocke/mo-di-diffusion

cd ~/stable-diffusion-webui
wget https://huggingface.co/nitrosocke/mo-di-diffusion/resolve/main/moDi-v1-pruned.ckpt -O ./models/Stable-diffusion/moDi-v1-pruned.ckpt
wget https://huggingface.co/runwayml/stable-diffusion-v1-5/resolve/main/v1-inference.yaml -O ./models/Stable-diffusion/moDi-v1-pruned.yaml

あとがき

管理人

気をつけよう著作権。