Merge pull request #4527 from d8ahazard/Accelerate

accelerate

basicsr

diffusers

fairscale==0.4.4

transformers==4.19.2

diffusers==0.3.0

accelerate==0.12.0

basicsr==1.4.2

gfpgan==1.3.8

gradio==3.9

set GIT=

set VENV_DIR=

set COMMANDLINE_ARGS=

set ACCELERATE=

call webui.bat

#export CODEFORMER_COMMIT_HASH=""

#export BLIP_COMMIT_HASH=""

# Uncomment to enable accelerated launch

#export ACCELERATE="True"

###########################################

: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.