Rook / gpt2config.json
gaur3009's picture
Create gpt2config.json
51fcf82 verified
raw
history blame contribute delete
718 Bytes
{
"_name_or_path": "gpt2",
"architectures": [
"GPT2LMHeadModel"
],
"model_type": "gpt2",
"vocab_size": 50257,
"n_positions": 1024,
"n_ctx": 1024,
"n_embd": 768,
"n_layer": 12,
"n_head": 12,
"activation_function": "gelu_new",
"initializer_range": 0.02,
"layer_norm_epsilon": 1e-05,
"resid_pdrop": 0.1,
"embd_pdrop": 0.1,
"attn_pdrop": 0.1,
"summary_type": "cls_index",
"summary_use_proj": true,
"summary_activation": null,
"summary_proj_to_labels": true,
"summary_first_dropout": 0.1,
"scale_attn_weights": true,
"use_cache": true,
"bos_token_id": 50256,
"eos_token_id": 50256,
"output_attentions": false,
"output_hidden_states": false,
"return_dict": true
}