File size: 1,092 Bytes
efab89c |
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 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 |
{
"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": [
[
3,
"allegro--klej-cdsc-e"
],
[
2,
"allegro--klej-psc"
],
[
2,
"allegro--klej-dyk"
],
[
5,
"PL-MTEB--scifield"
]
],
"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"
}
|