diff options
| author | DGdev91 <iperpido91@gmail.com> | 2023-04-05 23:36:27 +0000 |
|---|---|---|
| committer | DGdev91 <iperpido91@gmail.com> | 2023-04-05 23:36:27 +0000 |
| commit | 3a5b47e26e8cb1cd640fedfe7cb5263a588d7088 (patch) | |
| tree | 13878ec83e6d72402547cf66af2e7c3075fcb35e /webui.sh | |
| parent | 22bcc7be428c94e9408f589966c2040187245d81 (diff) | |
| download | stable-diffusion-webui-gfx803-3a5b47e26e8cb1cd640fedfe7cb5263a588d7088.tar.gz stable-diffusion-webui-gfx803-3a5b47e26e8cb1cd640fedfe7cb5263a588d7088.tar.bz2 stable-diffusion-webui-gfx803-3a5b47e26e8cb1cd640fedfe7cb5263a588d7088.zip | |
Forcing PyTorch version for AMD GPUs automatic install
The old code tries to install the newest versions of pytorch, wich is currently 2.0. Forcing it to 1.13.1
Diffstat (limited to 'webui.sh')
| -rwxr-xr-x | webui.sh | 2 |
1 files changed, 1 insertions, 1 deletions
@@ -118,7 +118,7 @@ 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" + export TORCH_COMMAND="pip install torch==1.13.1+rocm5.2 torchvision==0.14.1+rocm5.2 --extra-index-url https://download.pytorch.org/whl/rocm5.2" fi for preq in "${GIT}" "${python_cmd}" |