Merge pull request #30 from microsoft/hanyang/cpu-support

fix: missing device argument for cpu

src/mattersim/forcefield/potential.py

@@ -1117,7 +1117,7 @@ def __init__(
         """
         super().__init__(**kwargs)
         if potential is None:
-            self.potential = Potential.load()
+            self.potential = Potential.load(device=device)
         else:
             self.potential = potential
         self.compute_stress = compute_stress