Make model recognized as `text-generation`?

#1
by julien-c HF staff - opened
Files changed (1) hide show
  1. config.json +3 -7
config.json CHANGED
@@ -1,17 +1,13 @@
1
  {
2
- "_num_labels": 1,
3
  "activation_function": "gelu_new",
 
 
 
4
  "attn_pdrop": 0.1,
5
  "bos_token_id": 50256,
6
  "embd_pdrop": 0.1,
7
  "eos_token_id": 50256,
8
- "id2label": {
9
- "0": "LABEL_0"
10
- },
11
  "initializer_range": 0.02,
12
- "label2id": {
13
- "LABEL_0": 0
14
- },
15
  "layer_norm_epsilon": 1e-05,
16
  "model_type": "gpt2",
17
  "n_ctx": 1024,
1
  {
 
2
  "activation_function": "gelu_new",
3
+ "architectures": [
4
+ "GPT2LMHeadModel"
5
+ ],
6
  "attn_pdrop": 0.1,
7
  "bos_token_id": 50256,
8
  "embd_pdrop": 0.1,
9
  "eos_token_id": 50256,
 
 
 
10
  "initializer_range": 0.02,
 
 
 
11
  "layer_norm_epsilon": 1e-05,
12
  "model_type": "gpt2",
13
  "n_ctx": 1024,