Commit
·
dd5487b
1
Parent(s):
968ff65
remove mps device check
Browse files- event_handlers.py +3 -3
event_handlers.py
CHANGED
|
@@ -33,9 +33,9 @@ logger = logging.getLogger(__name__)
|
|
| 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 |
-
|
| 38 |
-
|
| 39 |
if torch.cuda.is_available():
|
| 40 |
devices.append("cuda")
|
| 41 |
best_device = "cuda"
|
|
|
|
| 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():
|
| 40 |
devices.append("cuda")
|
| 41 |
best_device = "cuda"
|