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 /webui-macos-env.sh | |
| 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
Diffstat (limited to 'webui-macos-env.sh')
| -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 |