config.json

{
    "model_type": "AutoencoderKL",
    "latent_channels": 4,
    "input_shape": [3, 256, 256],
    "normalization_type": "layer"
}