Merge pull request #9593 from gakada/tcmalloc (1ffb44b0) · Commits · github_fork / Stable Diffusion Webui

README.md

+1 −0

Original line number	Diff line number	Diff line
		@@ -120,6 +120,7 @@ sudo pacman -S wget git python3
		bash <(wget -qO- https://raw.githubusercontent.com/AUTOMATIC1111/stable-diffusion-webui/master/webui.sh)
		```
		3. Run `webui.sh`.
		4. Check `webui-user.sh` for options.
		### Installation on Apple Silicon

		Find the instructions [here](https://github.com/AUTOMATIC1111/stable-diffusion-webui/wiki/Installation-on-Apple-Silicon).

+3 −0

Original line number	Diff line number	Diff line
		@@ -43,4 +43,7 @@
		# Uncomment to enable accelerated launch
		#export ACCELERATE="True"

		# Uncomment to disable TCMalloc
		#export NO_TCMALLOC="True"

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

+18 −3

Original line number	Diff line number	Diff line
		@@ -172,15 +172,30 @@ else
		exit 1
		fi

		# Try using TCMalloc on Linux
		prepare_tcmalloc() {
		if [[ "${OSTYPE}" == "linux"* ]] && [[ -z "${NO_TCMALLOC}" ]] && [[ -z "${LD_PRELOAD}" ]]; then
		TCMALLOC="$(ldconfig -p \| grep -Po "libtcmalloc.so.\d" \| head -n 1)"
		if [[ ! -z "${TCMALLOC}" ]]; then
		echo "Using TCMalloc: ${TCMALLOC}"
		export LD_PRELOAD="${TCMALLOC}"
		else
		printf "\e[1m\e[31mCannot locate TCMalloc (improves CPU memory usage)\e[0m\n"
		fi
		fi
		}

		if [[ ! -z "${ACCELERATE}" ]] && [ ${ACCELERATE}="True" ] && [ -x "$(command -v accelerate)" ]
		then
		printf "\n%s\n" "${delimiter}"
		printf "Accelerating launch.py..."
		printf "\n%s\n" "${delimiter}"
		prepare_tcmalloc
		exec accelerate launch --num_cpu_threads_per_process=6 "${LAUNCH_SCRIPT}" "$@"
		else
		printf "\n%s\n" "${delimiter}"
		printf "Launching launch.py..."
		printf "\n%s\n" "${delimiter}"
		prepare_tcmalloc
		exec "${python_cmd}" "${LAUNCH_SCRIPT}" "$@"
		fi