support torch-roch-5.7

This commit is contained in:
Vladimir Mandic
2023-10-03 21:32:21 -04:00
parent 0303cfab28
commit 86d66b70d2
3 changed files with 4 additions and 6 deletions
+2 -1
View File
@@ -7,8 +7,8 @@
- Add FreeU for *backend:diffusers*
for *backend:original* use extension: <https://github.com/ljleb/sd-webui-freeu>
- Add HyperTile: <https://github.com/tfernd/HyperTile>
- Include chaiNNer extension as submodule
- Convert IPEX slicing to generic: <https://github.com/vladmandic/automatic/blob/dev/modules/intel/ipex/attention.py>
- 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**:
+1 -4
View File
@@ -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')