microsoft
/
DialoGPT-medium
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Update config.json

This commit is contained in:
system 2020-03-11 16:12:53 +00:00
parent 9aad4637a4
commit e9ce49e76b
1 changed files with 6 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
config.json
View File

@ -1,13 +1,16 @@
{ {
"architectures": [
"GPT2LMHeadModel"
],
"bos_token_id": 50256, "bos_token_id": 50256,
"eos_token_ids": [ "eos_token_ids": [
50256 50256
], ],
"initializer_range": 0.02, "initializer_range": 0.02,
"layer_norm_epsilon": 1e-05, "layer_norm_epsilon": 1e-05,
"model_type": "gpt2", "model_type": "gpt2",
"n_ctx": 1024, "n_ctx": 1024,
"n_embd": 768, "n_embd": 1024,
"n_head": 16, "n_head": 16,
"n_layer": 24, "n_layer": 24,
"n_positions": 1024, "n_positions": 1024,