Fix accelerate check when spaces in path
This commit is contained in:
parent
0a54bd3395
commit
ac6fd2a5d9
|
@ -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
|
||||
|
|
Loading…
Reference in New Issue