sam_vit_large / kf_config.json
IMvision12's picture
Upgrade kf_config.json to v2 (nested config + weight_dtype)
466a57a verified
Raw
History Blame Contribute Delete
702 Bytes
{
"library_name": "kerasformers",
"kerasformers_version": "1.2.1",
"model_module": "kerasformers.models.sam",
"model_class": "SAMPromptableSegment",
"variant": "sam_vit_large",
"weights": "model.weights.h5",
"schema_version": 2,
"weight_dtype": "float32",
"model_type": "sam",
"vision_config": {
"vision_hidden_size": 1024,
"vision_num_hidden_layers": 24,
"vision_num_attention_heads": 16,
"vision_mlp_dim": 4096,
"vision_global_attn_indexes": [
5,
11,
17,
23
],
"num_multimask_outputs": 3,
"multimask_output": true,
"enable_boxes": false,
"enable_masks": false,
"image_size": 1024
}
}