From 14486378f593f0cb60178374691832f1f3a566ab Mon Sep 17 00:00:00 2001 From: Ilyes Batatia <48651863+ilyes319@users.noreply.github.com> Date: Fri, 15 Dec 2023 13:32:56 +0000 Subject: [PATCH] update the small model to energy model --- mace/calculators/foundations_models.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/mace/calculators/foundations_models.py b/mace/calculators/foundations_models.py index 370ccb28..64e064ba 100644 --- a/mace/calculators/foundations_models.py +++ b/mace/calculators/foundations_models.py @@ -57,11 +57,10 @@ def mace_mp( elif model in (None, "small", "medium", "large") or str(model).startswith("https:"): try: urls = dict( - small="https://tinyurl.com/2ymut2zu", - medium="https://tinyurl.com/y7uhwpje", + small="https://tinyurl.com/2jmmb8b7", # 2023-12-10-mace-128-L0_energy_epoch-249.model + medium="https://tinyurl.com/y7uhwpje", # 2023-12-03-mace-128-L1_epoch-199.model large="https://figshare.com/ndownloader/files/43117273", ) - # default URL points to 2023-12-03-mace-128-L1_epoch-199.model checkpoint_url = ( urls.get(model, urls["medium"]) if model in (None, "small", "medium", "large")