config.json

{
  "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
  }
}