Spaces:
Running
on
Zero
Running
on
Zero
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"
|