diff --git a/config.json b/config.json new file mode 100644 index 0000000..2914d88 --- /dev/null +++ b/config.json @@ -0,0 +1,16 @@ +{ + "bos_token_id": 50256, + "eos_token_ids": [ + 50256 + ], + "initializer_range": 0.02, + "layer_norm_epsilon": 1e-05, + "model_type": "gpt2", + "n_ctx": 1024, + "n_embd": 768, + "n_head": 16, + "n_layer": 24, + "n_positions": 1024, + "pad_token_id": 50256, + "vocab_size": 50257 +}