Merge pull request #9404 from DGdev91/master (7b02b17a) · Commits · github_fork / Stable Diffusion Webui

webui.sh

+3 −2

Original line number	Diff line number	Diff line
		@@ -118,7 +118,8 @@ case "$gpu_info" in
		esac
		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"
		# AMD users will still use torch 1.13 because 2.0 does not seem to work.
		export TORCH_COMMAND="pip install torch==1.13.1+rocm5.2 torchvision==0.14.1+rocm5.2 --index-url https://download.pytorch.org/whl/rocm5.2"
		fi

		for preq in "${GIT}" "${python_cmd}"