config.json

{
  "img_size": 32,
  "patch_size": 4,
  "embed_dim": 128,
  "depth": 8,
  "num_heads": 4,
  "in_channels": 3,
  "model_type": "dvit"
}