File size: 2,103 Bytes
fa2ac91
 
 
 
 
 
 
 
 
 
 
 
 
 
53e5f6d
fa2ac91
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
{
  "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
}