conrad-nit-5.1-8B / config.json
deepconradlabs's picture
Fix model config for GPT2 checkpoint
135d327
raw
history blame contribute delete
369 Bytes
{
"model_type": "gpt2",
"architectures": [
"GPT2LMHeadModel"
],
"vocab_size": 50257,
"n_positions": 1024,
"n_ctx": 1024,
"n_embd": 2,
"n_layer": 2,
"n_head": 2,
"n_inner": 8,
"activation_function": "gelu_new",
"layer_norm_epsilon": 1e-5,
"initializer_range": 0.02,
"bos_token_id": 50256,
"eos_token_id": 50256,
"use_cache": true
}