Fix for non-AMD GPUs.

lshqqytiger · Feb 11, 2023 · f69f079 · f69f079
1 parent 72989e8
commit f69f079
Showing 1 changed file with 2 additions and 1 deletion.
diff --git 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: