diff options
author | brkirch <brkirch@users.noreply.github.com> | 2023-05-12 15:15:43 +0000 |
---|---|---|
committer | brkirch <brkirch@users.noreply.github.com> | 2023-05-12 15:15:43 +0000 |
commit | 0cab07b2f1078df29c7f3b52a3ae6b27d9ed0764 (patch) | |
tree | 7d56c4ad46d0923e95013e9ca7b7e07e2dc2e299 | |
parent | 5ab7f213bec2f816f9c5644becb32eb72c8ffb89 (diff) | |
download | stable-diffusion-webui-gfx803-0cab07b2f1078df29c7f3b52a3ae6b27d9ed0764.tar.gz stable-diffusion-webui-gfx803-0cab07b2f1078df29c7f3b52a3ae6b27d9ed0764.tar.bz2 stable-diffusion-webui-gfx803-0cab07b2f1078df29c7f3b52a3ae6b27d9ed0764.zip |
Set PyTorch version to 2.0.1 for macOS
-rw-r--r-- | webui-macos-env.sh | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/webui-macos-env.sh b/webui-macos-env.sh index 10ab81c9..6354e73b 100644 --- a/webui-macos-env.sh +++ b/webui-macos-env.sh @@ -11,7 +11,7 @@ fi export install_dir="$HOME" export COMMANDLINE_ARGS="--skip-torch-cuda-test --upcast-sampling --no-half-vae --use-cpu interrogate" -export TORCH_COMMAND="pip install torch torchvision" +export TORCH_COMMAND="pip install torch==2.0.1 torchvision==0.15.2" export K_DIFFUSION_REPO="https://github.com/brkirch/k-diffusion.git" export K_DIFFUSION_COMMIT_HASH="51c9778f269cedb55a4d88c79c0246d35bdadb71" export PYTORCH_ENABLE_MPS_FALLBACK=1 |