cyrusyc commited on
Commit
a002c28
2 Parent(s): 541da9c 707eeed

Merge branch 'main' of hf.co:cyrusyc/mace-universal

Browse files
Files changed (1) hide show
  1. README.md +3 -2
README.md CHANGED
@@ -30,7 +30,8 @@ atoms = Atoms("NaCl")
30
 
31
  calculator = MACECalculator(
32
  model_paths=/path/to/pretrained.model,
33
- device=device
 
34
  )
35
 
36
  atoms.calc = calculator
@@ -68,7 +69,7 @@ If you use the pretrained models in this repository, please cite all the followi
68
 
69
 
70
  @misc {yuan_chiang_2023,
71
- author = { {Yuan Chiang} },
72
  title = { mace-universal (Revision e5ebd9b) },
73
  year = 2023,
74
  url = { https://huggingface.co/cyrusyc/mace-universal },
 
30
 
31
  calculator = MACECalculator(
32
  model_paths=/path/to/pretrained.model,
33
+ device=device,
34
+ default_dtype="float32" or "float64",
35
  )
36
 
37
  atoms.calc = calculator
 
69
 
70
 
71
  @misc {yuan_chiang_2023,
72
+ author = { {Yuan Chiang, Philipp Benner} },
73
  title = { mace-universal (Revision e5ebd9b) },
74
  year = 2023,
75
  url = { https://huggingface.co/cyrusyc/mace-universal },