SpliNet / config.json
Angshul's picture
Upload repaired SpliNet 2B-token pretrained model
4ec5e47 verified
Raw History Blame Contribute Delete
1.04 kB
{
"architectures": [
"SpliNetForMaskedLM"
],
"auto_map": {
"AutoConfig": "configuration_splinet.SpliNetConfig",
"AutoModel": "modeling_splinet.SpliNetModel",
"AutoModelForMaskedLM": "modeling_splinet.SpliNetForMaskedLM"
},
"bos_token_id": 1,
"dtype": "float32",
"eos_token_id": 2,
"hidden_act": "gelu_new",
"hidden_dropout_prob": 0.1,
"hidden_size": 768,
"initializer_range": 0.02,
"intermediate_size": 3072,
"layer_norm_eps": 1e-12,
"max_position_embeddings": 512,
"model_type": "splinet",
"num_hidden_layers": 12,
"pad_token_id": 3,
"splinet_num_heads": 12,
"splinet_order": 2,
"splinet_radius": 16,
"splinet_sidedness": "single",
"tie_word_embeddings": true,
"tokenizer_class": "SpliNetTokenizer",
"tokenizer_sha256": "119ec6b2af9cbbc56f297bd606b69f79f6e3130a34ee56122ee813a58d15bb9d",
"tpu_short_seq_length": 512,
"training_tokens": 2000000000,
"transformers_version": "5.16.1",
"type_vocab_size": 4,
"use_tpu_fourier_optimizations": false,
"vocab_size": 32000
}