Optional Accelerate Launch (0a54bd33) · Commits · github_fork / Stable Diffusion Webui

requirements.txt

+1 −0

+1 −0

+1 −0

+3 −0

Original line number	Original line	Diff line number	Diff line
	@@ -40,4 +40,7 @@ export COMMANDLINE_ARGS=""
	#export CODEFORMER_COMMIT_HASH=""		#export CODEFORMER_COMMIT_HASH=""
	#export BLIP_COMMIT_HASH=""		#export BLIP_COMMIT_HASH=""

			# Uncomment to enable accelerated launch
			#export ACCELERATE="True"

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

+13 −0

Original line number	Original line	Diff line number	Diff line
	@@ -28,15 +28,28 @@ goto :show_stdout_stderr
	:activate_venv		:activate_venv
	set PYTHON="%~dp0%VENV_DIR%\Scripts\Python.exe"		set PYTHON="%~dp0%VENV_DIR%\Scripts\Python.exe"
	echo venv %PYTHON%		echo venv %PYTHON%
			if [%ACCELERATE%] == ["True"] goto :accelerate
	goto :launch		goto :launch

	:skip_venv		: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		:launch
	%PYTHON% launch.py %*		%PYTHON% launch.py %*
	pause		pause
	exit /b		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		:show_stdout_stderr

	echo.		echo.