add EquiformerV2 OC20 ckpt

mlip_arena/models/externals.py

@@ -173,6 +173,22 @@ class EquiformerV2(OCPCalculator):
             force=atoms.get_forces(),
         )
 
+class EquiformerV2OC20(OCPCalculator):
+    def __init__(
+        self,
+        model_name="EquiformerV2-31M-S2EF-OC20-All+MD",
+        local_cache="/tmp/ocp/",
+        cpu=False,
+        seed=0,
+        **kwargs,
+    ) -> None:
+        super().__init__(
+            model_name=model_name,
+            local_cache=local_cache,
+            cpu=cpu,
+            seed=0,
+            **kwargs,
+        )
 
 class eSCN(OCPCalculator):
     def __init__(

mlip_arena/models/registry.yaml

@@ -71,6 +71,24 @@ EquiformerV2(OC22):
   nvt: true
   npt: false
 
+EquiformerV2(OC20):
+  module: externals
+  class: EquiformerV2OC20
+  family: equiformer
+  username: cyrusyc
+  last-update: 2024-07-08T00:00:00
+  datetime: 2024-07-08T00:00:00
+  datasets:
+    - ocp
+  gpu-tasks:
+    - homonuclear-diatomics
+    - combustion
+  github: https://github.com/FAIR-Chem/fairchem
+  doi: https://arxiv.org/abs/2306.12059
+  prediction: EF
+  nvt: true
+  npt: false
+
 eSCN(OC20):
   module: externals
   class: eSCN