diff options
| author | DaniAndTheWeb <57776841+DaniAndTheWeb@users.noreply.github.com> | 2023-01-14 14:46:23 +0000 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2023-01-14 14:46:23 +0000 |
| commit | 6192a222bf7131771a2cd7655a64a5b24a1e6e2e (patch) | |
| tree | 77d1eaf6280e7a0d4c369e7e8277c20375899c30 /webui.sh | |
| parent | 629612935372aa7faeb764b5660431e49b93de24 (diff) | |
| download | stable-diffusion-webui-gfx803-6192a222bf7131771a2cd7655a64a5b24a1e6e2e.tar.gz stable-diffusion-webui-gfx803-6192a222bf7131771a2cd7655a64a5b24a1e6e2e.tar.bz2 stable-diffusion-webui-gfx803-6192a222bf7131771a2cd7655a64a5b24a1e6e2e.zip | |
Export TORCH_COMMAND for AMD from the webui
Diffstat (limited to 'webui.sh')
| -rwxr-xr-x | webui.sh | 1 |
1 files changed, 1 insertions, 0 deletions
@@ -168,6 +168,7 @@ else gpu_info=$(lspci | grep VGA) if echo "$gpu_info" | grep -q "AMD" then + export TORCH_COMMAND="pip install torch torchvision --extra-index-url https://download.pytorch.org/whl/rocm5.2" HSA_OVERRIDE_GFX_VERSION=10.3.0 exec "${python_cmd}" "${LAUNCH_SCRIPT}" "$@" else exec "${python_cmd}" "${LAUNCH_SCRIPT}" "$@" |