enable mac silicone acceleration

JudeWells · Sep 25, 2024 · ca0ef5d · ca0ef5d
1 parent 53abfe6
commit ca0ef5d
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/utils/common.py b/src/utils/common.py
@@ -37,8 +37,8 @@ def get_torch_device(force_cpu=False):
     try:
         if torch.cuda.is_available() and not force_cpu:
             device_string = "cuda"
-        # elif torch.backends.mps.is_available() and not force_cpu:
-        #     device_string = "mps"
+        elif torch.backends.mps.is_available() and not force_cpu:
+            device_string = "mps"
         else:
             device_string = "cpu"
     except Exception as exc: