{
  "in_channels": 3,
  "patch_size": 16,
  "model_dim": 896,
  "decoder_depth": 8,
  "decoder_start_blocks": 2,
  "decoder_end_blocks": 2,
  "bottleneck_dim": 128,
  "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,
  "flux2_batch_norm_eps": 0.0001
}