{
  "framework": "jax",
  "model_type": "canvit",
  "model_config": {
    "embed_dim": 768,
    "num_heads": 12,
    "n_blocks": 12,
    "patch_size": 16,
    "ffn_ratio": 4.0,
    "rope_base": 100.0,
    "layerscale_init": 1e-05,
    "rw_stride": 2,
    "enable_reads": true,
    "n_backbone_registers": 5,
    "n_canvas_registers": 16,
    "canvas_num_heads": 8,
    "canvas_head_dim": 128,
    "enable_vpe": true
  },
  "source": "canvit/canvitb16-add-vpe-pretrain-g128px-s512px-in21k-dv3b16-2026-02-02"
}