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