IMvision12's picture
Upgrade kf_config.json to v2 (nested config + weight_dtype)
f8d44c7 verified
Raw
History Blame Contribute Delete
773 Bytes
{
"library_name": "kerasformers",
"kerasformers_version": "1.2.1",
"model_module": "kerasformers.models.eomt",
"model_class": "EoMTUniversalSegment",
"variant": "eomt_large_coco_instance_640",
"weights": "model.weights.h5",
"schema_version": 2,
"weight_dtype": "float32",
"model_type": "eomt",
"vision_config": {
"hidden_dim": 1024,
"num_hidden_layers": 24,
"num_heads": 16,
"depths": 4,
"num_queries": 200,
"num_classes": 80,
"layerscale_value": 1e-05,
"patch_size": 16,
"num_register_tokens": 4,
"num_upscale_blocks": 2,
"mlp_ratio": 4,
"drop_path_rate": 0.0,
"attention_dropout": 0.0,
"use_swiglu_ffn": false,
"layer_norm_eps": 1e-06,
"image_size": 640
}
}