vae/config.json

{
  "_target": "hsigene.HSIGeneAutoencoderKL",
  "in_channels": 48,
  "out_channels": 48,
  "latent_channels": 96,
  "embed_dim": 4,
  "block_out_channels": [
    64,
    128,
    256
  ],
  "num_res_blocks": 4,
  "attn_resolutions": [
    16,
    32,
    64
  ],
  "dropout": 0.0,
  "double_z": true,
  "resolution": 256
}