diff --git a/modules/devices.py b/modules/devices.py index 356af1548e8..60dd108676c 100644 --- a/modules/devices.py +++ b/modules/devices.py @@ -2,7 +2,7 @@ import contextlib import torch import torch_directml -from modules import errors, atiadlxx +from modules import errors from modules.sd_hijack_utils import CondFunc from packaging import version from functools import reduce @@ -98,6 +98,7 @@ def enable_tf32(): try: dml = torch_directml.device(torch_directml.default_device()) if dml.type == "privateuseone" and "AMD" in torch_directml.device_name(dml.index): + from modules import atiadlxx adl = atiadlxx.ATIADLxx() hMEM = adl.getMemoryInfo2(0).iHyperMemorySize else: