fix typo

2020-03-12 21:10:51 +01:00
parent f1c71da115
commit 6a82f774f2
1 changed files with 1 additions and 1 deletions
--- a/tests/test_modeling_bart.py
+++ b/tests/test_modeling_bart.py
@@ -82,7 +82,7 @@ class ModelTester:
            dropout=self.hidden_dropout_prob,
            attention_dropout=self.attention_probs_dropout_prob,
            max_position_embeddings=self.max_position_embeddings,
-            eos_token_ids=[self.eos_token_id],
+            eos_token_ids=[2],
            bos_token_id=self.bos_token_id,
            pad_token_id=self.pad_token_id,
        )