File size: 1,283 Bytes
c9800e0 | 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 | {
"architectures": [
"MSAForCausalLM"
],
"attention_bias": false,
"attention_dropout": 0.0,
"bos_token_id": 151643,
"eos_token_id": 151645,
"head_dim": 128,
"hidden_act": "silu",
"hidden_size": 2560,
"initializer_range": 0.02,
"intermediate_size": 9728,
"max_position_embeddings": 262144,
"max_window_layers": 36,
"model_type": "msa",
"msa_config": {
"ansloss_weight": 1.0,
"aux_loss": false,
"aux_loss_method": "INFONCE_DECOUPLE",
"auxloss_weight": 0.1,
"chunk_reduce_method": "max",
"decouple_pooling_mode": "mean",
"decouple_router": true,
"head_reduce_method": "mean",
"infonce_loss_temp": 0.1,
"lmloss_weigth": 1.0,
"pooling_kernel_size": 64,
"query_reduce_method": "max",
"recloss_weight": 0.0,
"rewrite_position": true,
"router_layer_idx": "18,19,20,21,22,23,24,25,26,27,28,29,30,31,32,33,34,35",
"top_k_docs": 16
},
"num_attention_heads": 32,
"num_hidden_layers": 36,
"num_key_value_heads": 8,
"rms_norm_eps": 1e-06,
"rope_scaling": null,
"rope_theta": 5000000,
"sliding_window": false,
"tie_word_embeddings": true,
"torch_dtype": "bfloat16",
"transformers_version": "4.51.3",
"use_cache": false,
"use_sliding_window": false,
"vocab_size": 151936
}
|