voidful commited on
Commit
a37cfd1
1 Parent(s): 2cf1cb2

update config

Browse files
Files changed (1) hide show
  1. config.json +1 -21
config.json CHANGED
@@ -43,30 +43,10 @@
43
  "no_repeat_ngram_size": 3,
44
  "normalize_before": false,
45
  "normalize_embedding": true,
46
- "num_beams": 4,
47
  "num_hidden_layers": 6,
48
  "pad_token_id": 1,
49
  "scale_embedding": false,
50
- "task_specific_params": {
51
- "summarization": {
52
- "length_penalty": 1.0,
53
- "max_length": 128,
54
- "min_length": 12,
55
- "num_beams": 4
56
- },
57
- "summarization_cnn": {
58
- "length_penalty": 2.0,
59
- "max_length": 142,
60
- "min_length": 56,
61
- "num_beams": 4
62
- },
63
- "summarization_xsum": {
64
- "length_penalty": 1.0,
65
- "max_length": 62,
66
- "min_length": 11,
67
- "num_beams": 6
68
- }
69
- },
70
  "tie_word_embeddings": false,
71
  "transformers_version": "4.8.2",
72
  "use_cache": true,
 
43
  "no_repeat_ngram_size": 3,
44
  "normalize_before": false,
45
  "normalize_embedding": true,
46
+ "num_beams": 1,
47
  "num_hidden_layers": 6,
48
  "pad_token_id": 1,
49
  "scale_embedding": false,
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
50
  "tie_word_embeddings": false,
51
  "transformers_version": "4.8.2",
52
  "use_cache": true,