Commit ac6fd2a5 authored by d8ahazard's avatar d8ahazard

Fix accelerate check when spaces in path

parent 0a54bd33
...@@ -35,8 +35,8 @@ goto :launch ...@@ -35,8 +35,8 @@ goto :launch
:accelerate :accelerate
echo "Checking for accelerate" echo "Checking for accelerate"
dir %VENV_DIR%\Scripts\accelerate.exe >tmp/stdout.txt 2>tmp/stderr.txt set ACCELERATE="%~dp0%VENV_DIR%\Scripts\accelerate.exe"
if %ERRORLEVEL% == 0 goto :accelerate_launch if EXIST %ACCELERATE% goto :accelerate_launch
:launch :launch
%PYTHON% launch.py %* %PYTHON% launch.py %*
...@@ -44,8 +44,7 @@ pause ...@@ -44,8 +44,7 @@ pause
exit /b exit /b
:accelerate_launch :accelerate_launch
echo "Accelerating2" echo "Accelerating"
set ACCELERATE="%~dp0%VENV_DIR%\Scripts\accelerate.exe"
%ACCELERATE% launch --num_cpu_threads_per_process=6 launch.py %ACCELERATE% launch --num_cpu_threads_per_process=6 launch.py
pause pause
exit /b exit /b
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment