fix on torch_command + tested on debian

d62fbcc5 · orionaskatu · AUTOMATIC1111 · 7bf76af4 · d62fbcc5
Commit d62fbcc5 authored Sep 13, 2022 by orionaskatu Committed by AUTOMATIC1111 Sep 14, 2022
Hide whitespace changes
Inline Side-by-side

Showing with 6 additions and 0 deletions

webui.sh webui.sh +6 -0

No files found.
--- 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