vae/config.json

{
    "sample_size": 128,
    "in_channels": 3,
    "out_channels": 3,
    "down_block_types": [
        "DownEncoderBlock2D",
        "DownEncoderBlock2D",
        "DownEncoderBlock2D"
    ],
    "up_block_types": [
        "UpDecoderBlock2D",
        "UpDecoderBlock2D",
        "UpDecoderBlock2D"
    ],
    "block_out_channels": [128, 256, 512],
    "layers_per_block": 2,
    "latent_channels": 4,
    "scaling_factor": 0.18215
}