gruhit-patel commited on
Commit
7152f53
1 Parent(s): 858f052

Modified early stopping criteria

Browse files
Files changed (1) hide show
  1. QuoteGenerator.py +1 -1
QuoteGenerator.py CHANGED
@@ -33,7 +33,7 @@ class QuoteGenerator():
33
  top_k=top_k,
34
  top_p=top_p,
35
  do_sample = do_sample,
36
- early_stopping=True
37
  )
38
 
39
  return output[0]['generated_text']
 
33
  top_k=top_k,
34
  top_p=top_p,
35
  do_sample = do_sample,
36
+ early_stopping=num_beams > 1
37
  )
38
 
39
  return output[0]['generated_text']