PyTorch のアップデート【Stable Diffusion Web UI】

2023-05-022023-08-25

当ページのリンクには広告が含まれています。

こんにちは、ふじみやです。

以前、Stable Diffusion Web UI をローカル PC で使うための方法についてご紹介をしました。

あわせて読みたい

ローカル PC で AI を使って画像を生成しよう【Automatic1111】こんにちは、ふじみやです。最近 Stable Diffusion で自分好みの画像を生成するのにはまっています。以前、ローカル PC で画像生成をするアプリとして NMKD Stable Di…

しばらく使えていましたが、どうやらいつの間にか Stable Diffusion Web UI で使用されているプログラムのアップデートがされており、PyTorch の最新版（2.0.0）が必要になったようです。私の PC には PyTorch 1.12.1 がインストールされており、アップデートで動かなくなってしまっていました。

そこで、今回は PyTorch のバージョンアップをする方法についてご紹介をします。

PyTorch のバージョンアップ

仰々しく書いていますが、やり方は簡単。

通常は webui-user.bat をダブルクリックすることで Stable Diffusion Web UI を実行していると思いますが、中身を少し変えてあげるだけです。

@echo off

set PYTHON=
set GIT=
set VENV_DIR=
set COMMANDLINE_ARGS=--xformers --precision full --no-half
set PYTORCH_CUDA_ALLOC_CONF=garbage_collection_threshold:0.6, max_split_size_mb:128

call webui.bat

通常は上記の通りとなっていますが、COMMANDLINE_ARGSのところに--reinstall-torch と--reinstall-xformersを追記しましょう。

@echo off

set PYTHON=
set GIT=
set VENV_DIR=
set COMMANDLINE_ARGS=--xformers --precision full --no-half --reinstall-torch --reinstall-xformers
set PYTORCH_CUDA_ALLOC_CONF=garbage_collection_threshold:0.6, max_split_size_mb:128

call webui.bat

最後に、元のファイルを残しておく意味でファイル名を適当に変更して保存して、今後は作成したファイルを実行して Stable Diffusion Web UI を起動するようにしてください。

それではまた。