Add cbm, vbm models.

usnistgov · Apr 14, 2024 · 156dc6e · 156dc6e
1 parent e491c93
commit 156dc6e
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/alignn/pretrained.py b/alignn/pretrained.py
@@ -65,6 +65,18 @@
         "https://figshare.com/ndownloader/files/31458712",
         1,
     ],
+    "intermat_cbm": [
+        "https://figshare.com/ndownloader/files/45392908",
+        1,
+    ],
+    "intermat_vbm": [
+        "https://figshare.com/ndownloader/files/45392914",
+        1,
+    ],
+    "intermat_phi": [
+        "https://figshare.com/ndownloader/files/45392911",
+        1,
+    ],
     "jv_magmom_oszicar_alignn": [
         "https://figshare.com/ndownloader/files/31458685",
         1,