owlv2-large-patch14 / kf_config.json
IMvision12's picture
Upgrade kf_config.json to v2 (nested config + weight_dtype)
20560c5 verified
Raw
History Blame Contribute Delete
720 Bytes
{
"library_name": "kerasformers",
"kerasformers_version": "1.2.1",
"model_module": "kerasformers.models.owlv2",
"model_class": "Owlv2Detect",
"variant": "owlv2-large-patch14",
"weights": "model.weights.h5",
"schema_version": 2,
"weight_dtype": "float32",
"model_type": "owlv2",
"text_config": {
"hidden_dim": 768,
"intermediate_size": 3072,
"num_heads": 16,
"num_layers": 12,
"max_position_embeddings": 16,
"vocab_size": 49408
},
"vision_config": {
"image_size": 1008,
"patch_size": 14,
"hidden_dim": 1024,
"intermediate_size": 4096,
"num_layers": 24,
"num_heads": 16
},
"projection_dim": 768,
"image_size": null
}