Add instanovoplus-v1.1.0 config
Browse files- config.json +47 -25
config.json
CHANGED
|
@@ -1,42 +1,64 @@
|
|
| 1 |
{
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 2 |
"attention_type": "wavlm",
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 3 |
"cond_cross_attn_layers": [
|
| 4 |
0,
|
| 5 |
3,
|
| 6 |
6,
|
| 7 |
9
|
| 8 |
],
|
| 9 |
-
"cond_emb_dim": 768,
|
| 10 |
-
"conv_peak_encoder": false,
|
| 11 |
"conv_pos": 256,
|
| 12 |
"conv_pos_groups": 32,
|
| 13 |
-
"diffusion_s": 0.008,
|
| 14 |
"diffusion_type": "multinomial",
|
| 15 |
-
"
|
| 16 |
-
"
|
| 17 |
-
"
|
| 18 |
-
"
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 19 |
"isotope_error_range": [
|
| 20 |
0,
|
| 21 |
1
|
| 22 |
],
|
| 23 |
-
"layers": 12,
|
| 24 |
-
"max_charge": 10,
|
| 25 |
"max_length": 40,
|
| 26 |
-
"
|
| 27 |
-
|
| 28 |
-
|
| 29 |
-
|
| 30 |
-
|
| 31 |
-
|
| 32 |
-
|
| 33 |
-
|
| 34 |
-
|
| 35 |
-
|
| 36 |
-
|
| 37 |
-
|
| 38 |
-
|
| 39 |
-
|
| 40 |
-
|
| 41 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 42 |
}
|
|
|
|
| 1 |
{
|
| 2 |
+
"vocab_size": 33,
|
| 3 |
+
"layers": 12,
|
| 4 |
+
"dim": 768,
|
| 5 |
+
"nheads": 8,
|
| 6 |
+
"dropout": 0.1,
|
| 7 |
"attention_type": "wavlm",
|
| 8 |
+
"time_steps": 20,
|
| 9 |
+
"wav_encoder": "wavlm",
|
| 10 |
+
"wavlm_num_bucket": 140,
|
| 11 |
+
"wavlm_max_dist": 280,
|
| 12 |
+
"dim_feedforward": 1024,
|
| 13 |
+
"max_charge": 10,
|
| 14 |
+
"t_emb_dim": 768,
|
| 15 |
+
"t_emb_max_period": 10000,
|
| 16 |
+
"cond_emb_dim": 768,
|
| 17 |
+
"drop_cond_prob": 0.1,
|
| 18 |
"cond_cross_attn_layers": [
|
| 19 |
0,
|
| 20 |
3,
|
| 21 |
6,
|
| 22 |
9
|
| 23 |
],
|
|
|
|
|
|
|
| 24 |
"conv_pos": 256,
|
| 25 |
"conv_pos_groups": 32,
|
|
|
|
| 26 |
"diffusion_type": "multinomial",
|
| 27 |
+
"diffusion_s": 0.008,
|
| 28 |
+
"pos_encoding": "relative",
|
| 29 |
+
"use_flash_attention": false,
|
| 30 |
+
"conv_peak_encoder": false,
|
| 31 |
+
"n_peaks": 200,
|
| 32 |
+
"min_mz": 50.0,
|
| 33 |
+
"max_mz": 2500.0,
|
| 34 |
+
"min_intensity": 0.01,
|
| 35 |
+
"remove_precursor_tol": 2.0,
|
| 36 |
+
"precursor_mass_tol": 50,
|
| 37 |
"isotope_error_range": [
|
| 38 |
0,
|
| 39 |
1
|
| 40 |
],
|
|
|
|
|
|
|
| 41 |
"max_length": 40,
|
| 42 |
+
"diffusion_schedule": [
|
| 43 |
+
0.9959956407546997,
|
| 44 |
+
0.9899116158485413,
|
| 45 |
+
0.9837409257888794,
|
| 46 |
+
0.9774042963981628,
|
| 47 |
+
0.9708139896392822,
|
| 48 |
+
0.9638685584068298,
|
| 49 |
+
0.9564463496208191,
|
| 50 |
+
0.9483951926231384,
|
| 51 |
+
0.9395192861557007,
|
| 52 |
+
0.9295578598976135,
|
| 53 |
+
0.9181517958641052,
|
| 54 |
+
0.9047877192497253,
|
| 55 |
+
0.8887006044387817,
|
| 56 |
+
0.8686933517456055,
|
| 57 |
+
0.8427781462669373,
|
| 58 |
+
0.8073800206184387,
|
| 59 |
+
0.7553581595420837,
|
| 60 |
+
0.6700572371482849,
|
| 61 |
+
0.5015214085578918,
|
| 62 |
+
0.03162277489900589
|
| 63 |
+
]
|
| 64 |
}
|