stable-diffusion-webuiをWSL2で動かす - TadaoYamaokaの開発日記

この議事の内容は古いです。現在この手順では動きません。

Stable Diffusionが公開されて一週間もたたないうちに便利なUIがいくつもリリースされている。
その中でも、stable-diffusion-webuiが機能が充実している。
GitHub - hlky/stable-diffusion-webui: Stable Diffusion web UI

ということで、stable-diffusion-webuiをローカルPCのWindows11のWSL2上で動かせるようにした。

以下、構築手順を示す。

Conda環境とPyTorchの構築は、前回の記事の内容と同じである。
なお、後々バージョン不整合が起きたため、前回の記事の内容を修正している。

WSL2上にconda環境作成

WSL2のUbuntu 20.04上に環境構築する。

wget https://repo.anaconda.com/archive/Anaconda3-2022.05-Linux-x86_64.sh
sh Anaconda3-2022.05-Linux-x86_64.sh

conda環境作成

conda環境を作成する。

conda create -n stable-diffusion python=3.9
conda activate stable-diffusion

PyTorchインストール

PyTorchをインストールする。

conda install pytorch torchvision torchaudio cudatoolkit=11.6 -c pytorch -c conda-forge

Jupyterとpandas、matplotlibをインストールする。バージョン不整合を避けるためにconda-forgeからインストールする。

conda install jupyter pandas matplotlib -c conda-forge

必須ライブラリインストール

Stable Diffusionおよびstable-diffusion-webuiで必要となるライブラリをインストールする(scipyをcondaでインストールすると不整合が起きたのでpipでインストールしている)。

pip install diffusers transformers scipy ftfy invisible-watermark gradio pynvml omegaconf pytorch_lightning

pipにないパッケージインストール

k-diffusion

k-diffusionをgitからインストールする。

pip install git+https://github.com/crowsonkb/k-diffusion/

taming-transformers

taming-transformersをソースからインストールする(pipからインストールできるパッケージでは動作しない)。

cd ~
git clone https://github.com/CompVis/taming-transformers.git
cd taming-transformers
pip install -e .

stable-diffusionのソースインストール

cd ~
git clone https://github.com/CompVis/stable-diffusion.git

stable-diffusion-webuiのソースインストール

cd ~
git clone https://github.com/hlky/stable-diffusion-webui.git

stable-diffusionのソースディレクトリにコピーする。

cd stable-diffusion-webui
cp -r * ../stable-diffusion/

モデルダウンロード

Stable Diffusionのモデルをgithubからダウンロードする。

git lfsを使用するため、セットアップしていない場合は、セットアップする。

sudo apt install git-lfs
git lfs install

リポジトリをクローンする。

cd ~
git clone https://huggingface.co/CompVis/stable-diffusion-v-1-4-original

cloneにはHuggingfaceのアカウントが必要になる。

シンボリックリンク作成

モデルのチェックポイントのシンボリックリンクをstable-diffusionのソースディレクトリに作成する。

cd stable-diffusion/models/ldm/stable-diffusion-v1
ln -s ~/stable-diffusion-v-1-4-original/sd-v1-4.ckpt model.ckpt

stable-diffusionのパッケージインストール

cd ~
cd stable-diffusion
pip install -e .

stable-diffusion-webuiの起動

stable-diffusion-webuiを起動する。

python webui.py

初回は、各種ファイルのダウンロードが入るため、起動までに数分かかる。

Loading model from models/ldm/stable-diffusion-v1/model.ckpt
Global Step: 470000
LatentDiffusion: Running in eps-prediction mode
DiffusionWrapper has 859.52 M params.
making attention of type 'vanilla' with 512 in_channels
Working with z of shape (1, 4, 32, 32) = 4096 dimensions.
making attention of type 'vanilla' with 512 in_channels
Downloading vocab.json: 100%|█████████████████████████████████████████████| 939k/939k [00:00<00:00, 987kB/s]
Downloading merges.txt: 100%|█████████████████████████████████████████████| 512k/512k [00:00<00:00, 646kB/s]
Downloading special_tokens_map.json: 100%|██████████████████████████████████| 389/389 [00:00<00:00, 446kB/s]
Downloading tokenizer_config.json: 100%|████████████████████████████████████| 905/905 [00:00<00:00, 905kB/s]
Downloading config.json: 100%|█████████████████████████████████████████| 4.31k/4.31k [00:00<00:00, 3.62MB/s]
Downloading pytorch_model.bin: 100%|███████████████████████████████████| 1.59G/1.59G [00:32<00:00, 52.2MB/s]
/home/kei/anaconda3/envs/stable-diffusion/lib/python3.9/site-packages/gradio/deprecation.py:43: UserWarning: You have unused kwarg parameters in Button, please remove them: {'full_width': True}
  warnings.warn(
Running on local URL:  http://localhost:7860/

To create a public link, set `share=True` in `launch()`.