vae_config.json

{
  "_class_name": "MageVAE",
  "latent_channels": 128,
  "downsample_factor": 16,
  "sample_posterior": false
}