Commit
•
858f052
1
Parent(s):
1821ee5
Added early stopping to ensure quote stopping
Browse files- QuoteGenerator.py +2 -1
QuoteGenerator.py
CHANGED
|
@@ -32,7 +32,8 @@ class QuoteGenerator():
|
|
| 32 |
temperature=temperature,
|
| 33 |
top_k=top_k,
|
| 34 |
top_p=top_p,
|
| 35 |
-
do_sample = do_sample
|
|
|
|
| 36 |
)
|
| 37 |
|
| 38 |
return output[0]['generated_text']
|
|
|
|
| 32 |
temperature=temperature,
|
| 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']
|