Commit
·
968ff65
1
Parent(s):
134888b
update
Browse files- event_handlers.py +1 -1
event_handlers.py
CHANGED
@@ -33,7 +33,7 @@ logger = logging.getLogger(__name__)
|
|
33 |
def get_available_torch_devices() -> Tuple[List[str], str]:
|
34 |
devices = ["cpu"]
|
35 |
best_device = "cpu"
|
36 |
-
if torch.mps.is_available():
|
37 |
devices.append("mps")
|
38 |
best_device = "mps"
|
39 |
if torch.cuda.is_available():
|
|
|
33 |
def get_available_torch_devices() -> Tuple[List[str], str]:
|
34 |
devices = ["cpu"]
|
35 |
best_device = "cpu"
|
36 |
+
if torch.backend.mps.is_available():
|
37 |
devices.append("mps")
|
38 |
best_device = "mps"
|
39 |
if torch.cuda.is_available():
|