Public/automatic
1
0
Fork 0
mirror of https://github.com/vladmandic/automatic synced 2026-09-19 01:04:32 +02:00
Code Issues Packages Projects Releases Wiki Activity

rocm fix none error

Browse Source
This commit is contained in:
Seunghoon Lee
2024-09-26 13:25:06 +09:00
parent 220878cb03
commit 9ba7f4b1a0
1 changed files with 5 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
installer.py
+5 -4
View File
@@ -557,10 +557,11 @@ def install_rocm_zluda():
ort_package = os.environ.get('ONNXRUNTIME_PACKAGE', f"--pre onnxruntime-training{'' if ort_version is None else ('==' + ort_version)} --index-url https://pypi.lsh.sh/{rocm.version[0]}{rocm.version[2]} --extra-index-url https://pypi.org/simple")
install(ort_package, 'onnxruntime-training')
if 'Flash attention' in opts.get('sdp_options'):
install(rocm.get_flash_attention_command(device))
elif not args.experimental:
uninstall('flash-attn')
if device is not None:
if 'Flash attention' in opts.get('sdp_options'):
install(rocm.get_flash_attention_command(device))
elif not args.experimental:
uninstall('flash-attn')
if device is not None and rocm.version != "6.2" and rocm.version == rocm.version_torch and rocm.get_blaslt_enabled():
log.debug(f'ROCm hipBLASLt: arch={device.name} available={device.blaslt_supported}')