diff --git a/config.json b/config.json new file mode 100644 index 0000000..f4878c6 --- /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": 1280, + "n_head": 20, + "n_layer": 36, + "n_positions": 1024, + "pad_token_id": 50256, + "vocab_size": 50257 +}