diff options
| author | AUTOMATIC1111 <16777216c@gmail.com> | 2023-07-08 14:29:28 +0000 |
|---|---|---|
| committer | AUTOMATIC1111 <16777216c@gmail.com> | 2023-07-08 14:29:28 +0000 |
| commit | 4da92281f65a5d3620e61aef76dc2ec23394e706 (patch) | |
| tree | 560a66a9bc933c71913f4ac18720299035451444 /webui.sh | |
| parent | ed855783ed157136800869f609e5cf35072f78eb (diff) | |
| download | stable-diffusion-webui-gfx803-4da92281f65a5d3620e61aef76dc2ec23394e706.tar.gz stable-diffusion-webui-gfx803-4da92281f65a5d3620e61aef76dc2ec23394e706.tar.bz2 stable-diffusion-webui-gfx803-4da92281f65a5d3620e61aef76dc2ec23394e706.zip | |
pin version for torch for Navi3 according to comment from #11228
Diffstat (limited to 'webui.sh')
| -rwxr-xr-x | webui.sh | 2 |
1 files changed, 1 insertions, 1 deletions
@@ -134,7 +134,7 @@ case "$gpu_info" in *"Navi 2"*) export HSA_OVERRIDE_GFX_VERSION=10.3.0 ;; *"Navi 3"*) [[ -z "${TORCH_COMMAND}" ]] && \ - export TORCH_COMMAND="pip install --pre torch torchvision --index-url https://download.pytorch.org/whl/nightly/rocm5.5" + export TORCH_COMMAND="pip install --pre torch==2.1.0.dev-20230614+rocm5.5 torchvision==0.16.0.dev-20230614+rocm5.5 --index-url https://download.pytorch.org/whl/nightly/rocm5.5" # Navi 3 needs at least 5.5 which is only on the nightly chain ;; *"Renoir"*) export HSA_OVERRIDE_GFX_VERSION=9.0.0 |