From 62f718f4cb7310b2667b770dd06ea5b59dfcc70b Mon Sep 17 00:00:00 2001 From: Ryuhei Okuno <69704776+lil-lon@users.noreply.github.com> Date: Sat, 19 Sep 2026 14:10:12 +0900 Subject: [PATCH 1/2] Fix low_level_tutorial to load MACE-MPA-0 as documented --- examples/tutorials/low_level_tutorial.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/examples/tutorials/low_level_tutorial.py b/examples/tutorials/low_level_tutorial.py index c426a49cf..bbf037d05 100644 --- a/examples/tutorials/low_level_tutorial.py +++ b/examples/tutorials/low_level_tutorial.py @@ -67,7 +67,7 @@ # load mace_mp using the mace package loaded_model = mace_mp( - model="medium", + model="medium-mpa-0", return_raw_model=True, default_dtype=str(dtype).removeprefix("torch."), device=str(device), From 1b90bbfd239abb7ab13b8f068f38fcc3ba97dfd0 Mon Sep 17 00:00:00 2001 From: Ryuhei Okuno <69704776+lil-lon@users.noreply.github.com> Date: Sat, 19 Sep 2026 16:12:03 +0900 Subject: [PATCH 2/2] Restore MACE-MPA-0 in example scripts switched to MP-0 by #524 --- examples/benchmarking/scaling.py | 2 +- examples/scripts/1_introduction.py | 2 +- examples/scripts/2_structural_optimization.py | 2 +- examples/scripts/3_dynamics.py | 2 +- examples/scripts/5_workflow.py | 4 ++-- examples/scripts/6_phonons.py | 2 +- 6 files changed, 7 insertions(+), 7 deletions(-) diff --git a/examples/benchmarking/scaling.py b/examples/benchmarking/scaling.py index 612a7475c..2bf868b84 100644 --- a/examples/benchmarking/scaling.py +++ b/examples/benchmarking/scaling.py @@ -57,7 +57,7 @@ def load_mace_model(device: torch.device) -> MaceModel: """Load MACE model for benchmarking.""" loaded_model = mace_mp( - model="medium", + model="medium-mpa-0", return_raw_model=True, default_dtype="float64", device=str(device), diff --git a/examples/scripts/1_introduction.py b/examples/scripts/1_introduction.py index 9835a9f83..3f6025de7 100644 --- a/examples/scripts/1_introduction.py +++ b/examples/scripts/1_introduction.py @@ -126,7 +126,7 @@ # Load the raw model from the downloaded model loaded_model = mace_mp( - model="medium", + model="medium-mpa-0", return_raw_model=True, default_dtype=str(dtype).removeprefix("torch."), device=str(device), diff --git a/examples/scripts/2_structural_optimization.py b/examples/scripts/2_structural_optimization.py index 2d8eea3ac..adbbaf963 100644 --- a/examples/scripts/2_structural_optimization.py +++ b/examples/scripts/2_structural_optimization.py @@ -135,7 +135,7 @@ # Load MACE model loaded_model = mace_mp( - model="medium", + model="medium-mpa-0", return_raw_model=True, default_dtype=str(dtype).removeprefix("torch."), device=str(device), diff --git a/examples/scripts/3_dynamics.py b/examples/scripts/3_dynamics.py index aa1250298..39357a2d8 100644 --- a/examples/scripts/3_dynamics.py +++ b/examples/scripts/3_dynamics.py @@ -147,7 +147,7 @@ # Load MACE model loaded_model = mace_mp( - model="medium", + model="medium-mpa-0", return_raw_model=True, default_dtype=str(dtype).removeprefix("torch."), device=str(device), diff --git a/examples/scripts/5_workflow.py b/examples/scripts/5_workflow.py index 1b314e446..a0877b55f 100644 --- a/examples/scripts/5_workflow.py +++ b/examples/scripts/5_workflow.py @@ -49,7 +49,7 @@ log.info("Loading MACE model...") -mace = mace_mp(model="medium", return_raw_model=True) +mace = mace_mp(model="medium-mpa-0", return_raw_model=True) mace_model = MaceModel( model=mace, device=device, @@ -158,7 +158,7 @@ dtype_elastic = torch.float64 loaded_model = mace_mp( - model="medium", + model="medium-mpa-0", enable_cueq=False, device=str(device), default_dtype=str(dtype_elastic).removeprefix("torch."), diff --git a/examples/scripts/6_phonons.py b/examples/scripts/6_phonons.py index b38ac9c5f..5296163be 100644 --- a/examples/scripts/6_phonons.py +++ b/examples/scripts/6_phonons.py @@ -58,7 +58,7 @@ def require_not_none[T](value: T | None, message: str) -> T: # Load the MACE model loaded_model = mace_mp( - model="medium", + model="medium-mpa-0", return_raw_model=True, default_dtype=str(dtype).removeprefix("torch."), device=str(device),