decoder/geometry_config.json

{
  "coordinate_scale": 20.0,
  "distogram_bins": 32,
  "coordinate_mode": "free",
  "target_modules": [
    "q_proj",
    "k_proj",
    "v_proj",
    "o_proj",
    "in_proj_qkv",
    "in_proj_z",
    "in_proj_b",
    "in_proj_a",
    "out_proj",
    "gate_proj",
    "up_proj",
    "down_proj"
  ]
}