{
  "library_name": "zeromodels",
  "zeromodels_version": "1.2.1",
  "model_module": "zeromodels.models.pit",
  "model_class": "PiTImageClassify",
  "variant": "pit_b_distilled_224_in1k",
  "weights": "model.weights.h5",
  "schema_version": 2,
  "weight_dtype": "float32",
  "model_type": "pit",
  "vision_config": {
    "patch_size": 14,
    "stride": 7,
    "embed_dim": [
      256,
      512,
      1024
    ],
    "depth": [
      3,
      6,
      4
    ],
    "heads": [
      4,
      8,
      16
    ],
    "mlp_ratio": 4,
    "distilled": true,
    "image_size": 224,
    "num_classes": 1000
  }
}