PaulLerner commited on
Commit
52e2590
1 Parent(s): 9d6873c

fix: set num_beams to 1 (greedy search, tuned on inclure) for a better API usage

Browse files
Files changed (1) hide show
  1. config.json +1 -1
config.json CHANGED
@@ -45,7 +45,7 @@
45
  "no_repeat_ngram_size": 0,
46
  "normalize_before": true,
47
  "normalize_embedding": true,
48
- "num_beams": 4,
49
  "max_length": 1024,
50
  "num_hidden_layers": 6,
51
  "pad_token_id": 1,
 
45
  "no_repeat_ngram_size": 0,
46
  "normalize_before": true,
47
  "normalize_embedding": true,
48
+ "num_beams": 1,
49
  "max_length": 1024,
50
  "num_hidden_layers": 6,
51
  "pad_token_id": 1,