Create config.json

config.json

@@ -0,0 +1,16 @@
+{
+  "model_type": "ProtNHF",
+  "architectural_version": 1,
+  "dt": 0.05,
+  "niter": 4,
+  "hidden_dims": 128,
+  "std": 0.7,
+  "integrator": "leapfrog",
+  "n_types": 20,
+  "energy": {
+    "d_model": 320,
+    "ff_dim": 1280,
+    "n_heads": 20,
+    "n_layers": 6
+  }
+}