config.json

{
  "model_type": "divae",
  "image_size": 448,
  "patch_size": 16,
  "n_channels": 2,
  "enc_type": "vit_b_enc",
  "dec_type": "unet_patched",
  "quant_type": "fsq",
  "codebook_size": "8-8-8-6-5",
  "latent_dim": 5,
  "commitment_weight": 1.0,
  "post_mlp": true,
  "norm_codes": true,
  "num_train_timesteps": 1000,
  "beta_schedule": "linear",
  "prediction_type": "sample",
  "zero_terminal_snr": true,
  "modality": "ahn",
  "normalization": {
    "kind": "minmax",
    "clip_min": -20.0,
    "clip_max": 80.0
  },
  "geospatial_window_px": 448,
  "token_grid_size": 28,
  "vocab_size": 15360,
  "align_with_pixel_size_m": 0.08
}