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():
|