AITemplateでWaifu Diffusion 動かしてみた

まだ細かく検証できてないけど、動きます。
(AITemplate使わないバージョンとの比較まで出来てないですが。
Waifu Diffusion以外でもStableDiffusion派生のモデルなら動作するようです。

AITemplateの導入は下記を参照
※ 下記の内容が完了してる前提での手順になります。

Waifu DiffusionのモデルをDL

※ AITemplateのdocker上で作業

git lfs インストール

curl -s https://packagecloud.io/install/repositories/github/git-lfs/script.deb.sh | sudo bash
sudo apt-get install git-lfs
git lfs install

git clone 

cd /
git clone --depth 1  https://huggingface.co/hakurei/waifu-diffusion-v1-4

コンパイル

cd AITemplate/examples/05_stable_diffusion/
python3  scripts/compile.py --local-dir /waifu-diffusion # 落としてきたモデルのリポジトリのパスを指定

実行

python3 scripts/demo.py --local-dir /waifu-diffusion

# i2iのデモスクリプト
python3 scripts/demo_img2img.py --local-dir /waifu-diffusion

サンプルが、--local-dirを指定で、好きにモデルを切り替えられるようになってるので、Waifu Diffusion以外もいけます！
(Anything v4.0は試して無事動きました。