kf_config.json

{
  "library_name": "kerasformers",
  "kerasformers_version": "1.2.1",
  "model_module": "kerasformers.models.vit",
  "model_class": "ViTImageClassify",
  "variant": "vit_tiny_patch16_224_augreg_in21k",
  "weights": "model.weights.h5",
  "schema_version": 2,
  "weight_dtype": "float32",
  "model_type": "vit",
  "vision_config": {
    "patch_size": 16,
    "embed_dim": 192,
    "depth": 12,
    "num_heads": 3,
    "mlp_ratio": 4.0,
    "qkv_bias": true,
    "qk_norm": false,
    "image_size": 224,
    "num_classes": 21843
  }
}