Fix accelerate check when spaces in path

1 files changed, 3 insertions, 4 deletions

diff --git a/webui.bat b/webui.bat
index 2a066d60..d4d626e2 100644
--- a/webui.bat
+++ b/webui.bat
@@ -35,8 +35,8 @@ goto :launch
 

 :accelerate

 echo "Checking for accelerate"

-dir %VENV_DIR%\Scripts\accelerate.exe >tmp/stdout.txt 2>tmp/stderr.txt

-if %ERRORLEVEL% == 0 goto :accelerate_launch

+set ACCELERATE="%~dp0%VENV_DIR%\Scripts\accelerate.exe"

+if EXIST %ACCELERATE% goto :accelerate_launch

 

 :launch

 %PYTHON% launch.py %*

@@ -44,8 +44,7 @@ pause
 exit /b

 

 :accelerate_launch

-echo "Accelerating2"

-set ACCELERATE="%~dp0%VENV_DIR%\Scripts\accelerate.exe"

+echo "Accelerating"

 %ACCELERATE% launch --num_cpu_threads_per_process=6 launch.py

 pause

 exit /b