diff options
| author | AUTOMATIC1111 <16777216c@gmail.com> | 2023-04-29 11:09:51 +0000 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2023-04-29 11:09:51 +0000 |
| commit | 7b02b17a0151361278c9b833dfb80e7eb1af3741 (patch) | |
| tree | 696cd8e6a6f0991346194938c563eed340f003e5 | |
| parent | fdac48683595699189aa84f8c1bea7b3bf718801 (diff) | |
| parent | 967fb51df2cbe7ff6b8408c788c180b411e7f828 (diff) | |
| download | stable-diffusion-webui-gfx803-7b02b17a0151361278c9b833dfb80e7eb1af3741.tar.gz stable-diffusion-webui-gfx803-7b02b17a0151361278c9b833dfb80e7eb1af3741.tar.bz2 stable-diffusion-webui-gfx803-7b02b17a0151361278c9b833dfb80e7eb1af3741.zip | |
Merge pull request #9404 from DGdev91/master
Forcing PyTorch version for AMD GPUs automatic install
| -rwxr-xr-x | webui.sh | 5 |
1 files changed, 3 insertions, 2 deletions
@@ -118,8 +118,9 @@ 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" -fi + # 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}" do |