III-BERT / config.json
Naphon's picture
Update config.json
7450fab verified
Raw History Blame Contribute Delete
1.03 kB
{
"architectures": [
"IIIBERTForMaskedLM"
],
"attention_bias": false,
"auto_map": {
"AutoConfig": "iii_configuration.IIIBERTConfig",
"AutoModelForMaskedLM": "iii_modeling.IIIBERTForMaskedLM",
"AutoModelForSequenceClassification": "iii_modeling.IIIBERTForSequenceClassification",
"AutoModelForTokenClassification": "iii_modeling.IIIBERTForTokenClassification"
},
"bos_token_id": 24005,
"dropout": 0.1,
"dtype": "float32",
"embedding_dim": 128,
"eos_token_id": 6,
"hidden_dim": 768,
"initializer_range": 0.02,
"intermediate_dim": 2048,
"layers_diliation_size": [
1,
2,
4,
1
],
"layers_kernel_size": [
5,
5,
5,
0
],
"lora_dim": 192,
"max_sequence_length": 512,
"model_type": "iii_bert",
"n_heads": 6,
"n_layers": 6,
"n_sublayers": 4,
"pad_token_id": 5,
"pe_theta": 10000.0,
"rms_norm_eps": 1e-06,
"tie_word_embeddings": false,
"transformers_version": "5.14.1",
"vocab_size": 24008
}