config.json

{
  "in_channels": 3,
  "patch_size": 32,
  "model_dim": 1024,
  "encoder_depth": 8,
  "decoder_depth": 8,
  "decoder_start_blocks": 2,
  "decoder_end_blocks": 2,
  "bottleneck_dim": 384,
  "mlp_ratio": 4.0,
  "depthwise_kernel_size": 7,
  "adaln_low_rank_rank": 128,
  "bottleneck_posterior_kind": "diagonal_gaussian",
  "bottleneck_norm_mode": "disabled",
  "logsnr_min": -10.0,
  "logsnr_max": 10.0,
  "pixel_noise_std": 0.558,
  "latent_running_stats_eps": 0.0001
}