caixin1998 commited on
Commit
9f32655
1 Parent(s): cb83ff5

commit from caixin

Browse files
Files changed (1) hide show
  1. config.json +15 -3
config.json CHANGED
@@ -1,5 +1,4 @@
1
  {
2
- "_name_or_path": "poetry_gen_gpt.bin",
3
  "activation_function": "gelu_new",
4
  "architectures": [
5
  "GPT2LMHeadModel"
@@ -9,7 +8,13 @@
9
  "embd_pdrop": 0.1,
10
  "eos_token_id": 50256,
11
  "gradient_checkpointing": false,
 
 
 
12
  "initializer_range": 0.02,
 
 
 
13
  "layer_norm_epsilon": 1e-05,
14
  "model_type": "gpt2",
15
  "n_ctx": 1024,
@@ -18,12 +23,19 @@
18
  "n_inner": null,
19
  "n_layer": 12,
20
  "n_positions": 1024,
 
21
  "resid_pdrop": 0.1,
22
  "summary_activation": null,
23
  "summary_first_dropout": 0.1,
24
  "summary_proj_to_labels": true,
25
  "summary_type": "cls_index",
26
  "summary_use_proj": true,
27
- "use_cache": true,
 
 
 
 
 
 
28
  "vocab_size": 21128
29
- }
1
  {
 
2
  "activation_function": "gelu_new",
3
  "architectures": [
4
  "GPT2LMHeadModel"
8
  "embd_pdrop": 0.1,
9
  "eos_token_id": 50256,
10
  "gradient_checkpointing": false,
11
+ "id2label": {
12
+ "0": "LABEL_0"
13
+ },
14
  "initializer_range": 0.02,
15
+ "label2id": {
16
+ "LABEL_0": 0
17
+ },
18
  "layer_norm_epsilon": 1e-05,
19
  "model_type": "gpt2",
20
  "n_ctx": 1024,
23
  "n_inner": null,
24
  "n_layer": 12,
25
  "n_positions": 1024,
26
+ "output_past": true,
27
  "resid_pdrop": 0.1,
28
  "summary_activation": null,
29
  "summary_first_dropout": 0.1,
30
  "summary_proj_to_labels": true,
31
  "summary_type": "cls_index",
32
  "summary_use_proj": true,
33
+ "task_specific_params": {
34
+ "text-generation": {
35
+ "do_sample": true,
36
+ "max_length": 120
37
+ }
38
+ },
39
+ "tokenizer_class": "BertTokenizer",
40
  "vocab_size": 21128
41
+ }