kf_config.json

{
  "library_name": "kerasformers",
  "kerasformers_version": "1.2.1",
  "model_module": "kerasformers.models.sam",
  "model_class": "SAMPromptableSegment",
  "variant": "sam_vit_huge",
  "weights": "model.weights.json",
  "schema_version": 2,
  "weight_dtype": "float32",
  "model_type": "sam",
  "vision_config": {
    "vision_hidden_size": 1280,
    "vision_num_hidden_layers": 32,
    "vision_num_attention_heads": 16,
    "vision_mlp_dim": 5120,
    "vision_global_attn_indexes": [
      7,
      15,
      23,
      31
    ],
    "num_multimask_outputs": 3,
    "multimask_output": true,
    "enable_boxes": false,
    "enable_masks": false,
    "image_size": 1024
  }
}