{
  "act_layer": "SiLU",
  "decoder_depth": 12,
  "dim": 768,
  "domains_in": [
    "catalog",
    "tok_a_g",
    "tok_a_i",
    "tok_a_r",
    "tok_a_y",
    "tok_a_z",
    "tok_dec",
    "tok_ebv",
    "tok_flux_bp_gaia",
    "tok_flux_g",
    "tok_flux_g_gaia",
    "tok_flux_i",
    "tok_flux_r",
    "tok_flux_rp_gaia",
    "tok_flux_w1",
    "tok_flux_w2",
    "tok_flux_w3",
    "tok_flux_w4",
    "tok_flux_z",
    "tok_image",
    "tok_image_hsc",
    "tok_mag_g",
    "tok_mag_i",
    "tok_mag_r",
    "tok_mag_y",
    "tok_mag_z",
    "tok_parallax",
    "tok_ra",
    "tok_segmap",
    "tok_shape11",
    "tok_shape12",
    "tok_shape22",
    "tok_shape_e1",
    "tok_shape_e2",
    "tok_shape_r",
    "tok_spectrum_desi",
    "tok_spectrum_sdss",
    "tok_xp_bp",
    "tok_xp_rp",
    "tok_z"
  ],
  "domains_out": [
    "catalog",
    "tok_a_g",
    "tok_a_i",
    "tok_a_r",
    "tok_a_y",
    "tok_a_z",
    "tok_dec",
    "tok_ebv",
    "tok_flux_bp_gaia",
    "tok_flux_g",
    "tok_flux_g_gaia",
    "tok_flux_i",
    "tok_flux_r",
    "tok_flux_rp_gaia",
    "tok_flux_w1",
    "tok_flux_w2",
    "tok_flux_w3",
    "tok_flux_w4",
    "tok_flux_z",
    "tok_image",
    "tok_image_hsc",
    "tok_mag_g",
    "tok_mag_i",
    "tok_mag_r",
    "tok_mag_y",
    "tok_mag_z",
    "tok_parallax",
    "tok_ra",
    "tok_segmap",
    "tok_shape11",
    "tok_shape12",
    "tok_shape22",
    "tok_shape_e1",
    "tok_shape_e2",
    "tok_shape_r",
    "tok_spectrum_desi",
    "tok_spectrum_sdss",
    "tok_xp_bp",
    "tok_xp_rp",
    "tok_z"
  ],
  "encoder_depth": 12,
  "gated_mlp": true,
  "mlp_bias": false,
  "mlp_ratio": 4,
  "norm_bias": false,
  "num_heads": 12,
  "proj_bias": false,
  "qk_norm": true,
  "qkv_bias": false
}