dense/dit/config.json

{
  "model_type": "dense_dit",
  "model_class": "pixal3d.models.transformers.dense_dit.DenseDiTDenoiser",
  "resolution": 16,
  "in_channels": 8,
  "out_channels": 8,
  "model_channels": 1024,
  "cond_channels": 1024,
  "num_blocks": 24,
  "num_heads": 16,
  "mlp_ratio": 4,
  "patch_size": 1,
  "pe_mode": "ape",
  "qk_rms_norm": true,
  "use_checkpoint": true,
  "use_fp16": false,
  "latent_shape": [
    8,
    16,
    16,
    16
  ],
  "image_attn_mode": "proj"
}