add current changes

2020-03-06 15:14:43 +01:00
parent 333affcb81
commit 77e6775065
1 changed files with 2 additions and 1 deletions
--- 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