Commit ac6fd2a5 authored by d8ahazard's avatar d8ahazard
Browse files

Fix accelerate check when spaces in path

parent 0a54bd33
Loading
Loading
Loading
Loading
+3 −4
Original line number Diff line number Diff line
@@ -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