{
  "architectures": [
    "ESMFold2ExperimentalModel"
  ],
  "auto_map": {
    "AutoConfig": "modeling_fastplms.ESMFold2Config",
    "AutoModel": "modeling_fastplms.ESMFold2ExperimentalModel",
    "AutoModelForSequenceClassification": "modeling_fastplms.ESMFold2ExperimentalForSequenceClassification",
    "AutoModelForTokenClassification": "modeling_fastplms.ESMFold2ExperimentalForTokenClassification"
  },
  "confidence_head": {
    "distogram_bins": 128,
    "enabled": true,
    "folding_trunk": {
      "dropout": 0.25,
      "n_heads": 8,
      "n_layers": 4
    },
    "max_dist": 52.0,
    "min_dist": 2.0,
    "num_pae_bins": 64,
    "num_pde_bins": 64,
    "num_plddt_bins": 50
  },
  "d_pair": 256,
  "d_single": 384,
  "disable_msa_features": true,
  "dtype": "float32",
  "esmc_id": "Synthyra/ESMplusplus_small",
  "fastplms_checkpoint_hash": "bb2eaf6389e1d58f5117ca3173a5ee73d73548cda5c89f95f85c4a2280ad0e7e",
  "fastplms_checkpoint_repo_id": "Synthyra/ESMFold2-300",
  "fastplms_model_id": "esmfold2_300",
  "fastplms_runtime_bundle_sha256": "2ef7cccec8dfd129c6621d7bbb4b9f91a0ed1be1fd3f62d670259a4a9e199af0",
  "folding_trunk": {
    "dropout": 0.25,
    "n_heads": 8,
    "n_layers": 24
  },
  "force_lm_dropout_during_inference": false,
  "inputs": {
    "atom_encoder": {
      "d_atom": 128,
      "d_token": 768,
      "expansion_ratio": 2,
      "n_blocks": 3,
      "n_heads": 4,
      "n_spatial_rope_pairs_per_axis": 2,
      "n_uid_rope_pairs": 10,
      "spatial_rope_base_frequency": 20.0,
      "swa_window_size": 128,
      "uid_rope_base_frequency": 10000.0
    },
    "d_inputs": 451
  },
  "lm_d_model": 960,
  "lm_dropout": 0.0,
  "lm_num_layers": 30,
  "model_type": "esmfold2",
  "msa_conditioning": false,
  "msa_encoder": {
    "d_hidden": 32,
    "d_msa": 128,
    "enabled": false,
    "msa_head_width": 32,
    "n_heads_msa": 8,
    "n_layers": 4
  },
  "n_relative_chain_bins": 2,
  "n_relative_residx_bins": 32,
  "num_diffusion_samples": 16,
  "num_loops": 3,
  "structure_head": {
    "diffusion_module": {
      "atom_num_blocks": 3,
      "atom_num_heads": 4,
      "c_atom": 128,
      "c_s_inputs": 451,
      "c_token": 768,
      "c_z": 256,
      "fourier_dim": 256,
      "relpos_r_max": 32,
      "relpos_s_max": 2,
      "sigma_data": 16.0,
      "token_num_blocks": 12,
      "token_num_heads": 16,
      "transition_multiplier": 2
    },
    "distogram_bins": 128,
    "gamma_0": 0.605,
    "gamma_min": 1.107,
    "inference_num_steps": 15,
    "inference_p": 8.0,
    "inference_s_max": 160.0,
    "inference_s_min": 0.0004,
    "noise_scale": 0.901,
    "step_scale": 1.638,
    "train_noise_log_mean": -1.2,
    "train_noise_log_std": 1.5
  },
  "transformers_version": "4.57.6",
  "type": "experimental",
  "confidence_head_release": {
    "name": "v1",
    "head_sha256": "40fd7f3d82fcefe8ad20ab2b32a37a68a84b54a527a4bce6eb9437bad2b77e31",
    "base_weight_sha256": "44d6797c5efebf24753d502b40950e0874871c96ceea14f2d7f7e39cebac67fd",
    "training_url": "https://wandb.ai/lhallee/fastplms-confidence/runs/9558b6d23daf",
    "updates": 780,
    "evaluation_request_sha256": "be5fb9a4b65a0c2abfa040069ffd4344a9d220ef8e595711c4849a62401e281a"
  }
}