config.json

{
  "in_channels": 3,
  "patch_size": 16,
  "model_dim": 896,
  "encoder_depth": 4,
  "decoder_depth": 8,
  "decoder_start_blocks": 2,
  "decoder_end_blocks": 2,
  "bottleneck_dim": 96,
  "mlp_ratio": 4.0,
  "depthwise_kernel_size": 7,
  "adaln_low_rank_rank": 128,
  "logsnr_min": -10.0,
  "logsnr_max": 10.0,
  "pixel_noise_std": 0.558,
  "pdg_mask_ratio": 0.75
}