vae_config.json

{
  "image_size": 256,
  "in_channels": 3,
  "out_channels": 3,
  "hidden_channels": 128,
  "num_res_blocks": 3,
  "channel_multipliers": [
    1,
    2,
    2,
    4,
    4
  ],
  "attention_resolutions": [
    32
  ],
  "dropout": 0.0,
  "use_checkpoint": false,
  "use_fp16": false,
  "resblock_updown": true,
  "z_channels": 8,
  "double_z": true,
  "λ_recon": 1.0,
  "λ_kl": 1e-6,
  "λ_lpips": 1.0,
  "λ_id": 0.1,
  "λ_gan": 0.5,
  "gan_start_step": 50000
}