sleeper371 commited on
Commit
dd5487b
·
1 Parent(s): 968ff65

remove mps device check

Browse files
Files changed (1) hide show
  1. 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
- devices.append("mps")
38
- best_device = "mps"
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"