{
  "affinity_heads": [
    {
      "dropout": 0.25,
      "expansion": 2,
      "max_distance": 22.0,
      "num_conditioning_transitions": 2,
      "num_distogram_bins": 64,
      "num_pair_update_blocks": 8,
      "num_triangle_attention_heads": 4,
      "triangle_attention_head_size": 32
    },
    {
      "dropout": 0.25,
      "expansion": 2,
      "max_distance": 22.0,
      "num_conditioning_transitions": 2,
      "num_distogram_bins": 64,
      "num_pair_update_blocks": 4,
      "num_triangle_attention_heads": 4,
      "triangle_attention_head_size": 32
    }
  ],
  "architecture": "boltz2_affinity",
  "atom_feature_size": 388,
  "atom_hidden_size": 128,
  "atom_pair_hidden_size": 16,
  "confidence": {
    "max_distance": 22.0,
    "normalize_single_input": true,
    "num_distogram_bins": 64,
    "num_pae_bins": 64,
    "num_pair_attention_heads": 16,
    "num_pairformer_blocks": 8,
    "num_pde_bins": 64,
    "num_plddt_bins": 50,
    "num_triangle_attention_heads": 4,
    "pair_dropout": 0.25,
    "post_layer_norm": false,
    "triangle_attention_head_size": 32
  },
  "denoiser": {
    "align_reverse_diffusion": true,
    "atom_hidden_size": 128,
    "atom_pair_hidden_size": 16,
    "num_atom_attention_heads": 4,
    "num_atom_decoder_blocks": 3,
    "num_atom_encoder_blocks": 3,
    "num_conditioning_transitions": 2,
    "num_key_atoms": 128,
    "num_query_atoms": 32,
    "num_token_encoder_attention_heads": 16,
    "num_token_encoder_blocks": 24,
    "post_layer_norm": false,
    "sigma_data": 16.0
  },
  "fourier_embedding_size": 256,
  "input_embeddings": {
    "add_cyclic_flag": true,
    "add_method_conditioning": true,
    "add_modified_flag": true,
    "add_molecule_type_feature": true,
    "num_atom_encoder_attention_heads": 4,
    "num_atom_encoder_blocks": 3,
    "num_method_types": 12,
    "num_molecule_types": 4,
    "vocab_size": 72
  },
  "max_conditioning_cutoff": 20.0,
  "min_conditioning_cutoff": 4.0,
  "num_bond_types": 7,
  "num_distogram_bins": 64,
  "num_recycles": 6,
  "pair_size": 128,
  "predict_b_factor": false,
  "relative_position_size": 139,
  "sampler": {
    "churn_rate": 0.8,
    "max_noise_level": 160.0,
    "min_noise_level": 0.0001,
    "noise_scale": 1.003,
    "num_sampling_steps": 200,
    "rho": 7.0,
    "step_scale": 1.5
  },
  "single_size": 384,
  "template": null,
  "trunk": {
    "msa_attention_head_size": 32,
    "msa_dropout": 0.15,
    "msa_outer_product_hidden_size": 32,
    "msa_pair_dropout": 0.25,
    "msa_size": 64,
    "msa_triangle_attention_head_size": 32,
    "num_msa_attention_heads": 8,
    "num_msa_blocks": 4,
    "num_msa_triangle_attention_heads": 4,
    "num_pair_attention_heads": 16,
    "num_pairformer_blocks": 64,
    "num_trunk_triangle_attention_heads": 4,
    "post_layer_norm": false,
    "trunk_pair_dropout": 0.25,
    "trunk_triangle_attention_head_size": 32,
    "use_paired_feature": true
  }
}