vae/config.json

{
  "_class_name": "WorldEngineVAE",
  "_diffusers_version": "0.36.0.dev0",
  "auto_map": {
    "AutoModel": "ae_model.WorldEngineVAE"
  },
  "sample_size": [
    360,
    640
  ],
  "channels": 3,
  "latent_channels": 16,
  "encoder_ch_0": 64,
  "encoder_ch_max": 256,
  "encoder_blocks_per_stage": [
    1,
    1,
    1,
    1
  ],
  "decoder_ch_0": 128,
  "decoder_ch_max": 1024,
  "decoder_blocks_per_stage": [
    1,
    1,
    1,
    1
  ],
  "use_middle_block": false,
  "skip_logvar": false,
  "scale_factor": 1.0,
  "shift_factor": 0.0
}