Small reformat of the GPU check (48045545) · Commits · github_fork / Stable Diffusion Webui

webui.sh

+5 −5

Original line number	Diff line number	Diff line
		@@ -109,6 +109,10 @@ if echo "$gpu_info" \| grep -q "Navi"
		then
		export HSA_OVERRIDE_GFX_VERSION=10.3.0
		fi
		if echo "$gpu_info" \| grep -q "AMD" && [[ -z "${TORCH_COMMAND}" ]]
		then
		export TORCH_COMMAND="pip install torch torchvision --extra-index-url https://download.pytorch.org/whl/rocm5.2"
		fi

		for preq in "${GIT}" "${python_cmd}"
		do
		@@ -171,9 +175,5 @@ else
		printf "\n%s\n" "${delimiter}"
		printf "Launching launch.py..."
		printf "\n%s\n" "${delimiter}"
		if echo "$gpu_info" \| grep -q "AMD" && [[ -z "${TORCH_COMMAND}" ]]
		then
		export TORCH_COMMAND="pip install torch torchvision --extra-index-url https://download.pytorch.org/whl/rocm5.2"
		fi
		exec "${python_cmd}" "${LAUNCH_SCRIPT}" "$@"
		fi