codebook_config:
  freeze: true
  num_codes: 8192
  embed_dim: 13
  entropy_loss_weight: 0.1
  commitment_loss_weight: 0.25
encoder_config:
  freeze: true
decoder_config:
  input_dim: 128
  trunk:
    num_blocks: 4
    sequence_state_dim: 128
    pairwise_state_dim: 32
    max_recycles: 1
    structure_module:
      c_s: 128
      c_z: 32
      no_blocks: 8
      no_angles: 7
    gradient_checkpointing: false