vae/config.json

{
  "_class_name": "AutoencoderMageVAE",
  "_diffusers_version": "0.37.0",
  "latent_channels": 128,
  "downsample_factor": 16,
  "encoder_hidden_size": 384,
  "encoder_num_blocks": 21,
  "encoder_patch_size": 16,
  "encoder_head_size": 768,
  "encoder_num_head_blocks": 2,
  "decoder_hidden_size": 384,
  "decoder_hidden_size_x": 32,
  "decoder_num_blocks": 24,
  "decoder_num_cond_blocks": 21,
  "decoder_bottleneck_dim": 128,
  "decoder_patch_size": 16,
  "sample_posterior": false
}