Skip to content

Commit 0d8a405

Browse files
Add UMA-S-1.2.1 model configuration
Signed-off-by: Andrew S. Rosen <asrosen93@gmail.com>
1 parent 84d1ed1 commit 0d8a405

1 file changed

Lines changed: 1 addition & 0 deletions

File tree

src/matcalc/utils.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -82,6 +82,7 @@
8282
"MatterSim-v1.0.0-5M": {"provider": "mattersim", "load_path": "MatterSim-v1.0.0-5M.pth"},
8383
# FAIRChem (UMA family) — upstream uses ``uma-s-1p2`` style; we expose them
8484
# under the unified ``<Arch>-<Size>-<Version>`` form.
85+
"UMA-S-1.2.1": {"provider": "fairchem", "model": "uma-s-1p2p1", "task_name": "omat"},
8586
"UMA-S-1.2": {"provider": "fairchem", "model": "uma-s-1p2", "task_name": "omat"},
8687
"UMA-S-1.1": {"provider": "fairchem", "model": "uma-s-1p1", "task_name": "omat"},
8788
"UMA-M-1.1": {"provider": "fairchem", "model": "uma-m-1p1", "task_name": "omat"},

0 commit comments

Comments
 (0)