claudios commited on
Commit
fcd81d8
1 Parent(s): 8b158cf

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +1 -1
config.json CHANGED
@@ -3,7 +3,7 @@
3
  "_num_labels": 2,
4
  "activation_function": "gelu_new",
5
  "architectures": [
6
- "GPT2Model"
7
  ],
8
  "attn_pdrop": 0.1,
9
  "bos_token_id": 0,
 
3
  "_num_labels": 2,
4
  "activation_function": "gelu_new",
5
  "architectures": [
6
+ "GPT2LMHeadModel"
7
  ],
8
  "attn_pdrop": 0.1,
9
  "bos_token_id": 0,