From 86d66b70d2579d6f10b50dbe89e1300d3aa21024 Mon Sep 17 00:00:00 2001 From: Vladimir Mandic Date: Tue, 3 Oct 2023 21:32:21 -0400 Subject: [PATCH] support torch-roch-5.7 --- CHANGELOG.md | 3 ++- extensions-builtin/stable-diffusion-webui-rembg | 2 +- installer.py | 5 +---- 3 files changed, 4 insertions(+), 6 deletions(-) diff --git a/CHANGELOG.md b/CHANGELOG.md index 939913d6c..18e522f55 100644 --- a/CHANGELOG.md +++ b/CHANGELOG.md @@ -7,8 +7,8 @@ - Add FreeU for *backend:diffusers* for *backend:original* use extension: - Add HyperTile: -- Include chaiNNer extension as submodule - Convert IPEX slicing to generic: +- Fix override settings This is a big one, with some major changes and new functionality... And probably the biggest release since introduction of **Diffusers** @@ -132,6 +132,7 @@ Upgrades are still possible and supported, but above is recommended for best exp thanks @Disty0 @Nuullll - **AMD ROCm**: - updated installer to support detect `ROCm` *5.4/5.5/5.6/5.7* + - support for `torch-rocm-5.7` - **CUDA**: - testing moved to `torch` *2.2.0-dev/cu121* - **GC**: diff --git a/extensions-builtin/stable-diffusion-webui-rembg b/extensions-builtin/stable-diffusion-webui-rembg index 99fa213b8..543bc702b 160000 --- a/extensions-builtin/stable-diffusion-webui-rembg +++ b/extensions-builtin/stable-diffusion-webui-rembg @@ -1 +1 @@ -Subproject commit 99fa213b8daff7a6200a0149f2bef7a81b99b30a +Subproject commit 543bc702ba3b04877328bfb7cded5679597b627f diff --git a/installer.py b/installer.py index 6ed551124..9d19e8c35 100644 --- a/installer.py +++ b/installer.py @@ -406,12 +406,9 @@ def check_torch(): except Exception as e: log.debug(f'ROCm hipconfig failed: {e}') rocm_ver = None - if rocm_ver in ['5.5', '5.6']: + if rocm_ver in ['5.5', '5.6', '5.7']: # install torch nightly via torchvision to avoid wasting bandwidth when torchvision depends on torch from yesterday torch_command = os.environ.get('TORCH_COMMAND', f'torchvision --pre --index-url https://download.pytorch.org/whl/nightly/rocm{rocm_ver}') - elif rocm_ver in ['5.7']: - # there is no torch nightly for rocm 5.7 yet - torch_command = os.environ.get('TORCH_COMMAND', 'torchvision --pre --index-url https://download.pytorch.org/whl/nightly/rocm5.6') else: torch_command = os.environ.get('TORCH_COMMAND', 'torch==2.0.1 torchvision==0.15.2 --index-url https://download.pytorch.org/whl/rocm5.4.2') xformers_package = os.environ.get('XFORMERS_PACKAGE', 'none')