From d62fbcc5aa9381ea416f5997f291190c1e74bf29 Mon Sep 17 00:00:00 2001
From: orionaskatu <100234619+orionaskatu@users.noreply.github.com>
Date: Tue, 13 Sep 2022 17:48:10 +0200
Subject: [PATCH] fix on torch_command + tested on debian

---
 webui.sh | 6 ++++++
 1 file changed, 6 insertions(+)

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