sleeper371 commited on
Commit
968ff65
·
1 Parent(s): 134888b
Files changed (1) hide show
  1. 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():