config.json

{
    "model_type": "vit",
    "architectures": ["ViTForImageClassification"],
    "num_labels": 5,
    "image_size": 224
}