{
  "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_large",
  "fastplms_checkpoint_hash": "4e3baff4574120dfa58a25f34c910a05943e343ce63ad268775d2ce33b3b8c64",
  "fastplms_checkpoint_repo_id": "Synthyra/ESMFold2-600",
  "fastplms_model_id": "esmfold2_600",
  "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": 1152,
  "lm_dropout": 0.0,
  "lm_num_layers": 36,
  "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": "e84726a050722e1b722712c87d17a5388bd3699e2520e4a59abb1d828dfb8de7",
    "base_weight_sha256": "11a53c1b4700b6c62a5a464fc3ec7076160c19e8584157f88e13275116cfd602",
    "training_url": "https://wandb.ai/lhallee/fastplms-confidence/runs/820d2cfa56c0",
    "updates": 780,
    "evaluation_request_sha256": "1874e73d7e3acfc0766ff38b51d4eb09e735610c41cbe388b6b19388fa8c10db"
  }
}