1 files changed, 3 insertions, 1 deletions
diff --git a/webui.bat b/webui.bat
index 209d972b..42e7d517 100644
--- a/webui.bat
+++ b/webui.bat
@@ -3,7 +3,7 @@
 if not defined PYTHON (set PYTHON=python)
 if not defined VENV_DIR (set "VENV_DIR=%~dp0%venv")
 
-
+set SD_WEBUI_RESTART=tmp/restart
 set ERROR_REPORTING=FALSE
 
 mkdir tmp 2>NUL
@@ -51,12 +51,14 @@ if EXIST %ACCELERATE% goto :accelerate_launch
 
 :launch
 %PYTHON% launch.py %*
+if EXIST tmp/restart goto :skip_venv
 pause
 exit /b
 
 :accelerate_launch
 echo Accelerating
 %ACCELERATE% launch --num_cpu_threads_per_process=6 launch.py
+if EXIST tmp/restart goto :skip_venv
 pause
 exit /b