diff options
| author | AUTOMATIC1111 <16777216c@gmail.com> | 2023-05-27 16:56:18 +0000 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2023-05-27 16:56:18 +0000 |
| commit | 348abeb99d1f5fd81a12b28b757a41282c7fa863 (patch) | |
| tree | 22514b35c7ccea5df92416864d803bf8571fd251 /webui.sh | |
| parent | ba812b449534be49527b88e653a88ac1a1703d0e (diff) | |
| parent | bdc371983ea817547378fd2232ff0eb22bb315fa (diff) | |
| download | stable-diffusion-webui-gfx803-348abeb99d1f5fd81a12b28b757a41282c7fa863.tar.gz stable-diffusion-webui-gfx803-348abeb99d1f5fd81a12b28b757a41282c7fa863.tar.bz2 stable-diffusion-webui-gfx803-348abeb99d1f5fd81a12b28b757a41282c7fa863.zip | |
Merge pull request #10722 from maybe-hello-world/master
Download ROCm for AMD GPU only if NVIDIA is not presented
Diffstat (limited to 'webui.sh')
| -rwxr-xr-x | webui.sh | 7 |
1 files changed, 5 insertions, 2 deletions
@@ -124,9 +124,12 @@ case "$gpu_info" in *) ;; esac -if echo "$gpu_info" | grep -q "AMD" && [[ -z "${TORCH_COMMAND}" ]] +if ! echo "$gpu_info" | grep -q "NVIDIA"; then - export TORCH_COMMAND="pip install torch==2.0.1+rocm5.4.2 torchvision==0.15.2+rocm5.4.2 --index-url https://download.pytorch.org/whl/rocm5.4.2" + if echo "$gpu_info" | grep -q "AMD" && [[ -z "${TORCH_COMMAND}" ]] + then + export TORCH_COMMAND="pip install torch==2.0.1+rocm5.4.2 torchvision==0.15.2+rocm5.4.2 --index-url https://download.pytorch.org/whl/rocm5.4.2" + fi fi for preq in "${GIT}" "${python_cmd}" |