vae/config.json

{
  "_class_name": "AutoencoderKL",
  "block_out_channels": [128, 256, 512],
  "in_channels": 3,
  "out_channels": 3,
  "latent_channels": 4,
  "sample_size": 512,
  "scaling_factor": 0.18215
}