sam_vit_base / kf_config.json
IMvision12's picture
Upgrade kf_config.json to v2 (nested config + weight_dtype)
6e927d2 verified
Raw
History Blame Contribute Delete
698 Bytes
{
"library_name": "kerasformers",
"kerasformers_version": "1.2.1",
"model_module": "kerasformers.models.sam",
"model_class": "SAMPromptableSegment",
"variant": "sam_vit_base",
"weights": "model.weights.h5",
"schema_version": 2,
"weight_dtype": "float32",
"model_type": "sam",
"vision_config": {
"vision_hidden_size": 768,
"vision_num_hidden_layers": 12,
"vision_num_attention_heads": 12,
"vision_mlp_dim": 3072,
"vision_global_attn_indexes": [
2,
5,
8,
11
],
"num_multimask_outputs": 3,
"multimask_output": true,
"enable_boxes": false,
"enable_masks": false,
"image_size": 1024
}
}