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

corrections as specified in PR conversation

Browse Source
This commit is contained in:
resonantsky
2026-03-23 16:19:33 +02:00
parent 312d909739
commit 0ee33ea60e
2 changed files with 19 additions and 82 deletions
Download Patch File Download Diff File Expand all files Collapse all files
scripts/rocm/rocm_mgr.py
+7 -18
View File
@@ -11,24 +11,13 @@ from scripts.rocm import rocm_profiles # pylint: disable=no-name-in-module
def _check_rocm() -> bool:
try:
from modules import shared
if getattr(shared.cmd_opts, 'use_rocm', False):
return True
except Exception:
pass
try:
if installer.torch_info.get('type') == 'rocm':
return True
except Exception:
pass
try:
import torch
if hasattr(torch.version, 'hip') and torch.version.hip is not None:
return True
except Exception:
pass
return False
from modules import shared
if getattr(shared.cmd_opts, 'use_rocm', False):
return True
if installer.torch_info.get('type') == 'rocm':
return True
import torch # pylint: disable=import-outside-toplevel
return hasattr(torch.version, 'hip') and torch.version.hip is not None
is_rocm = _check_rocm()