From 0d8a405eb6e41399fa2f84a627e334006a48e80d Mon Sep 17 00:00:00 2001 From: "Andrew S. Rosen" Date: Wed, 12 Aug 2026 17:53:47 -0400 Subject: [PATCH] Add UMA-S-1.2.1 model configuration Signed-off-by: Andrew S. Rosen --- src/matcalc/utils.py | 1 + 1 file changed, 1 insertion(+) diff --git a/src/matcalc/utils.py b/src/matcalc/utils.py index 98ab6b8f..ee120c07 100644 --- a/src/matcalc/utils.py +++ b/src/matcalc/utils.py @@ -82,6 +82,7 @@ "MatterSim-v1.0.0-5M": {"provider": "mattersim", "load_path": "MatterSim-v1.0.0-5M.pth"}, # FAIRChem (UMA family) — upstream uses ``uma-s-1p2`` style; we expose them # under the unified ``--`` form. + "UMA-S-1.2.1": {"provider": "fairchem", "model": "uma-s-1p2p1", "task_name": "omat"}, "UMA-S-1.2": {"provider": "fairchem", "model": "uma-s-1p2", "task_name": "omat"}, "UMA-S-1.1": {"provider": "fairchem", "model": "uma-s-1p1", "task_name": "omat"}, "UMA-M-1.1": {"provider": "fairchem", "model": "uma-m-1p1", "task_name": "omat"},