vae/config.json

{
  "model": {
    "base_learning_rate": 1.5e-7,
    "target": "ldm.models.autoencoder.AutoencoderKL",
    "params": {
      "monitor": "val/rec_loss",
      "embed_dim": 4,
      "lossconfig": {
        "target": "ldm.modules.losses.LPIPSWithDiscriminator",
        "params": {
          "disc_start": 50001,
          "kl_weight": 0.000001,
          "disc_weight": 0.5
        }
      },
      "ddconfig": {
        "double_z": true,
        "z_channels": 4,
        "resolution": 256,
        "in_channels": 3,
        "out_ch": 3,
        "ch": 128,
        "ch_mult": [
          1,
          2,
          4,
          4
        ],
        "num_res_blocks": 2,
        "attn_resolutions": [],
        "dropout": 0
      }
    }
  }
}