pszemraj commited on
Commit
c739ce3
1 Parent(s): d096425

decrease num beams for API

Browse files
Files changed (1) hide show
  1. README.md +3 -2
README.md CHANGED
@@ -176,12 +176,13 @@ widget:
176
  example_title: Richard & Mortimer
177
  parameters:
178
  max_length: 64
179
- min_length: 8
180
  no_repeat_ngram_size: 3
181
  early_stopping: true
182
  length_penalty: 0.3
 
183
  encoder_no_repeat_ngram_size: 3
184
- num_beams: 2
185
  ---
186
 
187
  # pszemraj/pegasus-x-large-book-summary
 
176
  example_title: Richard & Mortimer
177
  parameters:
178
  max_length: 64
179
+ min_length: 4
180
  no_repeat_ngram_size: 3
181
  early_stopping: true
182
  length_penalty: 0.3
183
+ repetition_penalty: 3.5
184
  encoder_no_repeat_ngram_size: 3
185
+ num_beams: 1
186
  ---
187
 
188
  # pszemraj/pegasus-x-large-book-summary