adammoss commited on
Commit
98746e5
1 Parent(s): 94ef93c

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +1 -1
config.json CHANGED
@@ -3,7 +3,7 @@
3
  "GPTModel"
4
  ],
5
  "dropout": 0.1,
6
- "head_type": "pretrain_lm",
7
  "is_causal": true,
8
  "model_type": "gptmodel",
9
  "n_embd": 384,
 
3
  "GPTModel"
4
  ],
5
  "dropout": 0.1,
6
+ "head_type": "lm",
7
  "is_causal": true,
8
  "model_type": "gptmodel",
9
  "n_embd": 384,