gruhit-patel
commited on
Commit
•
7152f53
1
Parent(s):
858f052
Modified early stopping criteria
Browse files- 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=
|
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']
|