diff options
| author | AUTOMATIC1111 <16777216c@gmail.com> | 2022-11-19 10:49:44 +0300 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2022-11-19 10:49:44 +0300 |
| commit | 0d098e46568a268180974953fe2783fff1333040 (patch) | |
| tree | 2195d6ef8d1f29b65682e88a707fb5c19dec5734 /webui.bat | |
| parent | 98947d173e3f1667eba29c904f681047dea9de90 (diff) | |
| parent | ac6fd2a5d91ca1ee36d721ef483b101597d3ee25 (diff) | |
Merge pull request #4527 from d8ahazard/Accelerate
Optional Accelerate Launch
Diffstat (limited to 'webui.bat')
| -rw-r--r-- | webui.bat | 12 |
1 files changed, 12 insertions, 0 deletions
@@ -28,15 +28,27 @@ 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"
+set ACCELERATE="%~dp0%VENV_DIR%\Scripts\accelerate.exe"
+if EXIST %ACCELERATE% goto :accelerate_launch
+
:launch
%PYTHON% launch.py %*
pause
exit /b
+:accelerate_launch
+echo "Accelerating"
+%ACCELERATE% launch --num_cpu_threads_per_process=6 launch.py
+pause
+exit /b
+
:show_stdout_stderr
echo.
|