tarekziade commited on
Commit
65992e2
1 Parent(s): 5ab7833

Update generation_config.json

Browse files
Files changed (1) hide show
  1. generation_config.json +4 -5
generation_config.json CHANGED
@@ -3,13 +3,13 @@
3
  "temperature": 1.0,
4
  "top_k": 50,
5
  "top_p": 1.0,
6
- "num_beams": 1,
7
- "repetition_penalty": 1.0,
8
  "length_penalty": 1.0,
9
- "early_stopping": false,
10
  "max_length": 50,
11
  "min_length": 0,
12
- "no_repeat_ngram_size": 0,
13
  "bad_words_ids": null,
14
  "num_return_sequences": 1,
15
  "diversity_penalty": 0.0,
@@ -23,4 +23,3 @@
23
  "output_scores": false,
24
  "use_cache": true
25
  }
26
-
 
3
  "temperature": 1.0,
4
  "top_k": 50,
5
  "top_p": 1.0,
6
+ "num_beams": 2,
7
+ "repetition_penalty": 1.2,
8
  "length_penalty": 1.0,
9
+ "early_stopping": true,
10
  "max_length": 50,
11
  "min_length": 0,
12
+ "no_repeat_ngram_size": 2,
13
  "bad_words_ids": null,
14
  "num_return_sequences": 1,
15
  "diversity_penalty": 0.0,
 
23
  "output_scores": false,
24
  "use_cache": true
25
  }