mtzig's picture
Model save
a16714f verified
{
"architectures": [
"NanoGPT"
],
"bias": true,
"block_size": 256,
"dropout": 0.0,
"mlp_dim": 12,
"model_type": "nanogpt",
"n_embd": 2,
"n_head": 1,
"n_layer": 1,
"nonlinearity": "RELU",
"torch_dtype": "float32",
"transformers_version": "4.46.0",
"use_NoPE": true,
"use_layernorm": true,
"vocab_size": 14
}