add current changes

77e67750 · Patrick von Platen · 333affcb · 77e67750
Commit 77e67750 authored Mar 06, 2020 by Patrick von Platen
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 1 deletion

tests/test_modeling_bart.py tests/test_modeling_bart.py +2 -1

No files found.
--- a/tests/test_modeling_bart.py
+++ b/tests/test_modeling_bart.py
@@ -515,7 +515,8 @@ class BartModelIntegrationTest(unittest.TestCase):
            max_length=max_length + 2,
            min_length=min_length + 1,
            no_repeat_ngram_size=3,
-            do_sample=False
+            do_sample=False,
+            early_stopping=True
        )
        decoded = [
            tok.decode(g, skip_special_tokens=True, clean_up_tokenization_spaces=False) for g in hypotheses_batch