diff options
| author | d8ahazard <d8ahazard@gmail.com> | 2022-11-09 17:15:17 +0000 |
|---|---|---|
| committer | d8ahazard <d8ahazard@gmail.com> | 2022-11-09 17:15:17 +0000 |
| commit | 0a54bd3395e30bf7772ad6f11d69c5c437a1ab73 (patch) | |
| tree | 909d4203ca95f68001965f19b9397da288bb38b1 /webui.bat | |
| parent | ac085628540d0ec6a988fad93f5b8f2154209571 (diff) | |
| download | stable-diffusion-webui-gfx803-0a54bd3395e30bf7772ad6f11d69c5c437a1ab73.tar.gz stable-diffusion-webui-gfx803-0a54bd3395e30bf7772ad6f11d69c5c437a1ab73.tar.bz2 stable-diffusion-webui-gfx803-0a54bd3395e30bf7772ad6f11d69c5c437a1ab73.zip | |
Optional Accelerate Launch
Diffstat (limited to 'webui.bat')
| -rw-r--r-- | webui.bat | 13 |
1 files changed, 13 insertions, 0 deletions
@@ -28,15 +28,28 @@ goto :show_stdout_stderr :activate_venv
set PYTHON="%~dp0%VENV_DIR%\Scripts\Python.exe"
echo venv %PYTHON%
+if [%ACCELERATE%] == ["True"] goto :accelerate
goto :launch
:skip_venv
+:accelerate
+echo "Checking for accelerate"
+dir %VENV_DIR%\Scripts\accelerate.exe >tmp/stdout.txt 2>tmp/stderr.txt
+if %ERRORLEVEL% == 0 goto :accelerate_launch
+
:launch
%PYTHON% launch.py %*
pause
exit /b
+:accelerate_launch
+echo "Accelerating2"
+set ACCELERATE="%~dp0%VENV_DIR%\Scripts\accelerate.exe"
+%ACCELERATE% launch --num_cpu_threads_per_process=6 launch.py
+pause
+exit /b
+
:show_stdout_stderr
echo.
|