voidful commited on
Commit
421706d
1 Parent(s): 26e2a81

Remove summarization parameters and add text-generation parameter

Browse files
Files changed (1) hide show
  1. config.json +3 -17
config.json CHANGED
@@ -48,23 +48,9 @@
48
  "pad_token_id": 0,
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,
 
48
  "pad_token_id": 0,
49
  "scale_embedding": false,
50
  "task_specific_params": {
51
+ "text-generation": {
52
+ "do_sample": true,
53
+ "max_length": 200
 
 
 
 
 
 
 
 
 
 
 
 
 
 
54
  }
55
  },
56
  "tie_word_embeddings": false,