Joemgu commited on
Commit
03c0d05
1 Parent(s): d9d6577

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +14 -1
config.json CHANGED
@@ -30,6 +30,19 @@
30
  "tie_word_embeddings": false,
31
  "torch_dtype": "float32",
32
  "transformers_version": "4.30.0",
33
- "use_cache": false,
 
 
 
 
 
 
 
 
 
 
 
 
 
34
  "vocab_size": 256384
35
  }
 
30
  "tie_word_embeddings": false,
31
  "torch_dtype": "float32",
32
  "transformers_version": "4.30.0",
33
+ "task_specific_params": {
34
+ "summarization": {
35
+ "early_stopping": true,
36
+ "length_penalty": 2.0,
37
+ "repetition_penalty": 3.5,
38
+ "max_length": 256,
39
+ "min_length": 16,
40
+ "encoder_no_repeat_ngram_size": 5,
41
+ "no_repeat_ngram_size": 4,
42
+ "num_beams": 4,
43
+ "prefix": "Write a title and summarize: "
44
+ }
45
+ },
46
+ "use_cache": true,
47
  "vocab_size": 256384
48
  }