inference_config.json

{
  "ATOMS_TO_LOAD": "backbone",
  "PARSE_PROTEIN": 1,
  "PARSE_DNA": 1,
  "PARSE_RNA": 1,
  "PARSE_RNA_AS_DNA": 0,
  "NA_SHARED_TOKENS": 1,
  "PROTEIN_BACKBONE_OCC_CUTOFF": 0.8,
  "PROTEIN_SIDE_CHAIN_OCC_CUTOFF": 0.5,
  "DNA_BACKBONE_OCC_CUTOFF": 0.8,
  "DNA_SIDE_CHAIN_OCC_CUTOFF": 0.5,
  "RNA_BACKBONE_OCC_CUTOFF": 0.8,
  "RNA_SIDE_CHAIN_OCC_CUTOFF": 0.5,
  "CROP_LARGE_STRUCTURES": 1,
  "BATCH_TOKENS": 6000,
  "NA_REF_ATOM": "C1'",
  "EXCLUDE_RES": ["HOH", "NA", "CL", "K", "BR"],
  "HIDDEN_DIM": 128,
  "NUM_ENCODER_LAYERS": 3,
  "NUM_DECODER_LAYERS": 3,
  "NUM_NEIGHBORS": 32,
  "INCLUDE_PRED_NA_N": 1,
  "USE_SEQUENCE_CONTEXT": true,
  "VOCAB_SIZE": 33,
  "NUM_LETTERS": 33
}