config.json

{
  "architectures": [
    "EsmForProteinFolding"
  ],
  "attention_probs_dropout_prob": 0.0,
  "classifier_dropout": null,
  "emb_layer_norm_before": false,
  "esmfold_config": {
    "bypass_lm": false,
    "embed_aa": true,
    "esm_ablate_pairwise": false,
    "esm_ablate_sequence": false,
    "esm_input_dropout": 0,
    "esm_type": "esm2_3B",
    "fp16_esm": false,
    "lddt_head_hid_dim": 128,
    "trunk": {
      "chunk_size": null,
      "cpu_grad_checkpoint": false,
      "dropout": 0,
      "layer_drop": 0,
      "max_recycles": 4,
      "num_blocks": 48,
      "pairwise_head_width": 32,
      "pairwise_state_dim": 128,
      "position_bins": 32,
      "sequence_head_width": 32,
      "sequence_state_dim": 1024,
      "structure_module": {
        "dropout_rate": 0.1,
        "epsilon": 1e-08,
        "inf": 100000.0,
        "ipa_dim": 16,
        "num_angles": 7,
        "num_blocks": 8,
        "num_heads_ipa": 12,
        "num_qk_points": 4,
        "num_resnet_blocks": 2,
        "num_transition_layers": 1,
        "num_v_points": 8,
        "pairwise_dim": 128,
        "resnet_dim": 128,
        "sequence_dim": 384,
        "trans_scale_factor": 10
      }
    },
    "use_esm_attn_map": false
  },
  "hidden_act": "gelu",
  "hidden_dropout_prob": 0.0,
  "hidden_size": 2560,
  "initializer_range": 0.02,
  "intermediate_size": 10240,
  "is_folding_model": true,
  "layer_norm_eps": 1e-05,
  "mask_token_id": 32,
  "max_position_embeddings": 1026,
  "model_type": "esm",
  "num_attention_heads": 40,
  "num_hidden_layers": 36,
  "pad_token_id": 1,
  "position_embedding_type": "rotary",
  "token_dropout": true,
  "torch_dtype": "float32",
  "transformers_version": "4.25.0.dev0",
  "use_cache": true,
  "vocab_list": [
    "<cls>",
    "<pad>",
    "<eos>",
    "<unk>",
    "L",
    "A",
    "G",
    "V",
    "S",
    "E",
    "R",
    "T",
    "I",
    "D",
    "P",
    "K",
    "Q",
    "N",
    "F",
    "Y",
    "M",
    "H",
    "W",
    "C",
    "X",
    "B",
    "U",
    "Z",
    "O",
    ".",
    "-",
    "<null_1>",
    "<mask>"
  ],
  "vocab_size": 33
}