|
@@ -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
|