diff --git a/config.json b/config.json new file mode 100644 index 0000000..d9ff4d1 --- /dev/null +++ b/config.json @@ -0,0 +1,14 @@ +{ + "afn": "gelu", + "attn_pdrop": 0.1, + "embd_pdrop": 0.1, + "initializer_range": 0.02, + "n_ctx": 512, + "n_embd": 768, + "n_head": 12, + "n_layer": 12, + "n_positions": 512, + "n_special": 0, + "resid_pdrop": 0.1, + "vocab_size": 40478 +}