From 6a82f774f257f60aa7dc8b813e90e3ffb60e32d1 Mon Sep 17 00:00:00 2001 From: Patrick von Platen Date: Thu, 12 Mar 2020 21:10:51 +0100 Subject: [PATCH] fix typo --- tests/test_modeling_bart.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/test_modeling_bart.py b/tests/test_modeling_bart.py index a0ad29830d..31a3a8a25a 100644 --- 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, )