{
  "attention_heads": 64,
  "dropout": 0.1,
  "embedding_dim": 128,
  "encoder_layers": 15,
  "encoder_width": 512,
  "ffn_width": 2048,
  "gradient_checkpointing": true,
  "max_molecule_atoms": 256,
  "max_pocket_atoms": 256,
  "molecule_vocab_size": 31,
  "pocket_vocab_size": 10
}