File size: 895 Bytes
7d677df |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 |
{
"architectures": [
"ILKTModel"
],
"auto_map": {
"AutoConfig": "config.ILKTConfig",
"AutoModel": "model.ILKTModel"
},
"backbone_config": {
"pretrained_model_name_or_path": "google-bert/bert-base-multilingual-cased",
"torch_dtype": "bfloat16",
"trust_remote_code": true
},
"cls_head_config": {
"dropout": 0.0,
"n_dense": 1,
"pool_type": "cls",
"use_batch_norm": true,
"use_layer_norm": false
},
"cls_heads": [],
"embedding_head_config": {
"dropout": 0.0,
"n_dense": 1,
"normalize_embeddings": false,
"pool_type": "cls",
"use_batch_norm": false,
"use_layer_norm": false
},
"hidden_size": 768,
"mlm_head_config": {
"dropout": 0.0,
"n_dense": 1,
"use_batch_norm": false,
"use_layer_norm": true
},
"model_type": "ILKT",
"torch_dtype": "float32",
"transformers_version": "4.41.2"
}
|