config.json

{
  "ch": 128,
  "ch_mult": [
    1,
    2,
    4
  ],
  "final_sigmoid_activation": true,
  "image_size": 512,
  "in_cam_params": 7,
  "in_channels": 3,
  "inject_noise_in_decoder": false,
  "num_res_blocks": 2,
  "out_channels": 3,
  "vit_depth": 12,
  "vit_hidden_size": 768,
  "vit_mlp_ratio": 4.0,
  "vit_num_heads": 12,
  "vit_patch_size": 1,
  "vit_use_checkpoint": false,
  "vit_use_qknorm": false,
  "vit_use_rmsnorm": true,
  "vit_use_rope": false,
  "vit_use_swiglu": true,
  "vit_wo_shift": false
}