config.json

{
  "vocab_size": 33,
  "layers": 12,
  "dim": 768,
  "nheads": 8,
  "dropout": 0.1,
  "attention_type": "wavlm",
  "time_steps": 20,
  "wav_encoder": "wavlm",
  "wavlm_num_bucket": 140,
  "wavlm_max_dist": 280,
  "dim_feedforward": 1024,
  "max_charge": 10,
  "t_emb_dim": 768,
  "t_emb_max_period": 10000,
  "cond_emb_dim": 768,
  "drop_cond_prob": 0.1,
  "cond_cross_attn_layers": [
    0,
    3,
    6,
    9
  ],
  "conv_pos": 256,
  "conv_pos_groups": 32,
  "diffusion_type": "multinomial",
  "diffusion_s": 0.008,
  "pos_encoding": "relative",
  "use_flash_attention": false,
  "conv_peak_encoder": false,
  "n_peaks": 200,
  "min_mz": 50.0,
  "max_mz": 2500.0,
  "min_intensity": 0.01,
  "remove_precursor_tol": 2.0,
  "precursor_mass_tol": 50,
  "isotope_error_range": [
    0,
    1
  ],
  "max_length": 40,
  "diffusion_schedule": [
    0.9959956407546997,
    0.9899116158485413,
    0.9837409257888794,
    0.9774042963981628,
    0.9708139896392822,
    0.9638685584068298,
    0.9564463496208191,
    0.9483951926231384,
    0.9395192861557007,
    0.9295578598976135,
    0.9181517958641052,
    0.9047877192497253,
    0.8887006044387817,
    0.8686933517456055,
    0.8427781462669373,
    0.8073800206184387,
    0.7553581595420837,
    0.6700572371482849,
    0.5015214085578918,
    0.03162277489900589
  ]
}