fix on torch_command + tested on debian
This commit is contained in:
parent
7bf76af40a
commit
d62fbcc5aa
1 changed files with 6 additions and 0 deletions
6
webui.sh
6
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
|
||||
|
||||
|
|
Loading…
Reference in a new issue