aboutsummaryrefslogtreecommitdiffstats
path: root/webui.sh
diff options
context:
space:
mode:
authororionaskatu <100234619+orionaskatu@users.noreply.github.com>2022-09-13 15:48:10 +0000
committerAUTOMATIC1111 <16777216c@gmail.com>2022-09-14 07:45:25 +0000
commitd62fbcc5aa9381ea416f5997f291190c1e74bf29 (patch)
tree5d30dcf5f10a9bd098e5b7f8c3cac74b82b2eaf3 /webui.sh
parent7bf76af40a59cfc63aa91f573ce1c8335a415a2b (diff)
downloadstable-diffusion-webui-gfx803-d62fbcc5aa9381ea416f5997f291190c1e74bf29.tar.gz
stable-diffusion-webui-gfx803-d62fbcc5aa9381ea416f5997f291190c1e74bf29.tar.bz2
stable-diffusion-webui-gfx803-d62fbcc5aa9381ea416f5997f291190c1e74bf29.zip
fix on torch_command + tested on debian
Diffstat (limited to 'webui.sh')
-rw-r--r--webui.sh6
1 files changed, 6 insertions, 0 deletions
diff --git a/webui.sh b/webui.sh
index 2773b517..cc4b81f7 100644
--- a/webui.sh
+++ b/webui.sh
@@ -41,6 +41,12 @@ then
venv_dir="venv"
fi
+# install command for torch
+if [[ -z "${TORCH_COMMAND}" ]]
+then
+ export TORCH_COMMAND=(python3 -m pip install torch==1.12.1+cu113 --extra-index-url https://download.pytorch.org/whl/cu113)
+fi
+
# Do not reinstall existing pip packages on Debian/Ubuntu
export PIP_IGNORE_INSTALLED=0