{
  "architecture": "pxdesign",
  "atom_feature_size": 385,
  "condition_embeddings": {
    "add_feature_size": 4,
    "atom_hidden_size": 128,
    "atom_pair_hidden_size": 16,
    "design_feature_size": 1,
    "hotspot_size": 1,
    "num_atom_encoder_attention_heads": 4,
    "num_atom_encoder_blocks": 3,
    "num_key_atoms": 128,
    "num_query_atoms": 32,
    "plddt_size": 1,
    "token_hidden_size": 384,
    "vocab_size": 72
  },
  "denoiser": {
    "atom_hidden_size": 128,
    "atom_pair_hidden_size": 16,
    "num_atom_decoder_attention_heads": 4,
    "num_atom_decoder_blocks": 4,
    "num_atom_encoder_attention_heads": 4,
    "num_atom_encoder_blocks": 4,
    "num_key_atoms": 128,
    "num_query_atoms": 32,
    "num_token_encoder_attention_heads": 16,
    "num_token_encoder_blocks": 16,
    "sigma_data": 16.0,
    "token_hidden_size": 768
  },
  "num_template_bins": 64,
  "pair_size": 128,
  "sampler": {
    "churn_rate": 1.0,
    "max_noise_level": 160.0,
    "max_step_scale": 2.5,
    "min_churn_noise_level": 0.01,
    "min_noise_level": 0.0004,
    "min_step_scale": 1.0,
    "noise_scale": 1.003,
    "rho": 7.0,
    "step_scale_schedule": "piecewise_65"
  },
  "single_size": 384,
  "token_feature_size": 449
}