Skip to content

Commit

Permalink
Merge pull request #293 from ACEsuit/develop
Browse files Browse the repository at this point in the history
edit readme example for foundation models
  • Loading branch information
ilyes319 authored Jan 12, 2024
2 parents a2b6e8e + 7532bfb commit 44cc610
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -186,8 +186,8 @@ from mace.calculators import mace_mp
from ase import build

atoms = build.molecule('H2O')
calc = mace_mp(model="medium", device='cuda')
atoms.set_calculator(calc)
calc = mace_mp(model="medium", dispersion=False, default_dtype="float32", device='cuda')
atoms.calc = calc
print(atoms.get_potential_energy())
```

Expand All @@ -204,7 +204,7 @@ from ase import build

atoms = build.molecule('H2O')
calc = mace_off(model="medium", device='cuda')
atoms.set_calculator(calc)
atoms.calc = calc
print(atoms.get_potential_energy())
```

Expand Down

0 comments on commit 44cc610

Please sign in to comment.