{
  "decoder_config": {
    "folding_trunk": {
      "chunk_size": null,
      "cpu_grad_checkpoint": false,
      "dropout": 0.1,
      "layer_drop": 0,
      "max_recycles": 4,
      "num_blocks": 32,
      "pairwise_head_width": 32,
      "pairwise_state_dim": 128,
      "position_bins": 32,
      "sequence_head_width": 32,
      "sequence_state_dim": 768,
      "structure_module": {
        "c_ipa": 16,
        "c_resnet": 128,
        "c_s": 384,
        "c_z": 128,
        "dropout_rate": 0.1,
        "epsilon": 1e-08,
        "inf": 100000.0,
        "no_angles": 7,
        "no_blocks": 8,
        "no_heads_ipa": 12,
        "no_qk_points": 4,
        "no_resnet_blocks": 2,
        "no_transition_layers": 1,
        "no_v_points": 8,
        "trans_scale_factor": 10
      }
    },
    "lddt_head_hidden_dim": 128,
    "quantize_dim": 384
  },
  "encoder_config": {
    "edge_emb": {
      "bessels": {
        "bessel_const": 40.0,
        "d": 128
      },
      "d": 128,
      "edge_type_emb": {
        "d": 128,
        "dropout": 0.1,
        "max_res_offset": 32,
        "max_sym_offset": 2
      },
      "k_for_knn": 30
    },
    "eqnet": {
      "attn_dropout": 0,
      "d": 128,
      "d_mult": 3,
      "dropout": 0.1,
      "n_head": 4
    },
    "n_eqnet": 12,
    "node_emb": {
      "d": 128,
      "dropout": 0.1,
      "mask_node_type": true,
      "n_chain_type": 100,
      "n_node_type": 200
    },
    "quantize": {
      "decay": 0.99,
      "dim": 384,
      "eps": 1e-05,
      "n_embed": 512,
      "normalize": true,
      "restart": 100,
      "usage_threshold": 1e-09
    }
  },
  "frozen_codebook": false
}