fmilletari-czi commited on
Commit
b732f7a
·
verified ·
1 Parent(s): f53abaa

Upload folder using huggingface_hub

Browse files
Files changed (2) hide show
  1. config.json +98 -0
  2. model.safetensors +3 -0
config.json ADDED
@@ -0,0 +1,98 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "architectures": [
3
+ "ESMFold2Model"
4
+ ],
5
+ "confidence_head": {
6
+ "distogram_bins": 39,
7
+ "folding_trunk": {
8
+ "dropout": 0.25,
9
+ "n_heads": 8,
10
+ "n_layers": 4
11
+ },
12
+ "max_dist": 50.75,
13
+ "min_dist": 3.25,
14
+ "num_pae_bins": 64,
15
+ "num_pde_bins": 64,
16
+ "num_plddt_bins": 50
17
+ },
18
+ "d_pair": 256,
19
+ "d_single": 384,
20
+ "disable_msa_features": false,
21
+ "dtype": "float32",
22
+ "folding_trunk": {
23
+ "dropout": 0.25,
24
+ "n_heads": 8,
25
+ "n_layers": 24
26
+ },
27
+ "inputs": {
28
+ "atom_encoder": {
29
+ "d_atom": 128,
30
+ "d_token": 768,
31
+ "expansion_ratio": 2,
32
+ "n_blocks": 3,
33
+ "n_heads": 4,
34
+ "n_spatial_rope_pairs_per_axis": 2,
35
+ "n_uid_rope_pairs": 10,
36
+ "spatial_rope_base_frequency": 20.0,
37
+ "swa_window_size": 128,
38
+ "uid_rope_base_frequency": 10000.0
39
+ },
40
+ "d_inputs": 451
41
+ },
42
+ "lm_d_model": 2560,
43
+ "lm_encoder": {
44
+ "enabled": true,
45
+ "n_layers": 4
46
+ },
47
+ "lm_num_layers": 80,
48
+ "model_type": "esmfold2_v2",
49
+ "msa_encoder": {
50
+ "d_hidden": 32,
51
+ "d_msa": 128,
52
+ "enabled": false,
53
+ "msa_head_width": 32,
54
+ "n_heads_msa": 8,
55
+ "n_layers": 4
56
+ },
57
+ "msa_encoder_overwrite": true,
58
+ "n_relative_chain_bins": 2,
59
+ "n_relative_residx_bins": 32,
60
+ "num_diffusion_samples": 32,
61
+ "num_recycles": 3,
62
+ "parcae": {
63
+ "coda_n_layers": 2,
64
+ "enabled": true,
65
+ "max_steps": 6,
66
+ "min_steps": 1,
67
+ "poisson_mean": 3.0
68
+ },
69
+ "structure_head": {
70
+ "diffusion_module": {
71
+ "atom_num_blocks": 3,
72
+ "atom_num_heads": 4,
73
+ "c_atom": 128,
74
+ "c_s_inputs": 451,
75
+ "c_token": 768,
76
+ "c_z": 256,
77
+ "fourier_dim": 256,
78
+ "relpos_r_max": 32,
79
+ "relpos_s_max": 2,
80
+ "sigma_data": 16.0,
81
+ "token_num_blocks": 12,
82
+ "token_num_heads": 16,
83
+ "transition_multiplier": 2
84
+ },
85
+ "distogram_bins": 64,
86
+ "gamma_0": 0.8,
87
+ "gamma_min": 1.0,
88
+ "inference_num_steps": 14,
89
+ "inference_p": 7.0,
90
+ "inference_s_max": 160.0,
91
+ "inference_s_min": 0.0004,
92
+ "noise_scale": 1.003,
93
+ "step_scale": 1.5,
94
+ "train_noise_log_mean": -1.2,
95
+ "train_noise_log_std": 1.5
96
+ },
97
+ "transformers_version": "4.57.6"
98
+ }
model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:60ca19f2898188beba92944365f7b909efd9c99212f5018af75cc47cd9a6184a
3
+ size 755416924