generator/config.json

{
  "vocab_size": 21,
  "edge_feature_size": 128,
  "hidden_size": 128,
  "num_encoder_layers": 3,
  "num_decoder_layers": 3,
  "k_neighbors": 48,
  "coord_noise_scale": 0.0,
  "dropout": 0.0,
  "node_feature_size": 3,
  "checkpoint_variant": "per_residue_label_membrane_mpnn_v_48_020",
  "noise_level": 0.2
}