Change default device to cpu

ElliottKasoar · Mar 1, 2024 · 658c9c2 · 658c9c2
1 parent 1e54720
commit 658c9c2
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/janus_core/mlip_calculators.py b/janus_core/mlip_calculators.py
@@ -16,7 +16,7 @@
 
 def choose_calculator(
     architecture: Literal[architectures] = "mace",
-    device: Literal[devices] = "cuda",
+    device: Literal[devices] = "cpu",
     **kwargs,
 ) -> Calculator:
     """
@@ -27,7 +27,7 @@ def choose_calculator(
     architecture : Literal[architectures], optional
         MLIP architecture. Default is "mace".
     device : Literal[devices]
-        Device to run calculator on. Default is "cuda".
+        Device to run calculator on. Default is "cpu".
     **kwargs
         Additional keyword arguments passed to the selected calculator.