config.json

{
  "architectures": ["VisionTransformer"],
  "model_type": "vit",
  "auto_map": {
    "AutoConfig": "vision_transformer.VisionTransformerConfig",
    "AutoModel": "vision_transformer.VisionTransformer"
  },
  "img_size": 224,
  "patch_size": 16,
  "in_chans": 1,
  "num_classes": 0,
  "embed_dim": 384,
  "depth": 12,
  "num_heads": 6,
  "mlp_ratio": 4.0,
  "qkv_bias": true,
  "qk_scale": null,
  "drop_rate": 0.0,
  "attn_drop_rate": 0.0,
  "drop_path_rate": 0.0,
  "torch_dtype": "float32"
}