diff options
| author | Vladimir Mandic <mandic00@live.com> | 2023-03-30 10:57:54 -0400 |
|---|---|---|
| committer | Vladimir Mandic <mandic00@live.com> | 2023-03-30 10:57:54 -0400 |
| commit | d5063e07e8b4737621978feffd37b18077b9ea64 (patch) | |
| tree | 1bfa91530f0074419b906cdd6af43a27c5110a52 /webui-macos-env.sh | |
| parent | 22bcc7be428c94e9408f589966c2040187245d81 (diff) | |
update torch
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 37cac4fb..65d80413 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==1.12.1 torchvision==0.13.1" +export TORCH_COMMAND="pip install torch torchvision --extra-index-url https://download.pytorch.org/whl/cu118" export K_DIFFUSION_REPO="https://github.com/brkirch/k-diffusion.git" export K_DIFFUSION_COMMIT_HASH="51c9778f269cedb55a4d88c79c0246d35bdadb71" export PYTORCH_ENABLE_MPS_FALLBACK=1 |