clip_vit_base_16 / kf_config.json
IMvision12's picture
Upgrade kf_config.json to v2 (nested config + weight_dtype)
bc8418a verified
Raw
History Blame Contribute Delete
700 Bytes
{
"library_name": "kerasformers",
"kerasformers_version": "1.2.1",
"model_module": "kerasformers.models.clip",
"model_class": "CLIPZeroShotClassify",
"variant": "clip_vit_base_16",
"weights": "model.weights.h5",
"schema_version": 2,
"weight_dtype": "float32",
"model_type": "clip",
"text_config": {
"hidden_dim": 512,
"num_heads": 8,
"num_layers": 12,
"mlp_ratio": 4.0,
"vocab_size": 49408,
"max_seq_len": 77
},
"vision_config": {
"num_layers": 12,
"hidden_dim": 768,
"patch_size": 16,
"mlp_ratio": 4.0,
"image_size": 224
},
"embed_dim": 512,
"hidden_act": "quick_gelu",
"layer_norm_eps": 1e-05
}