This commit is contained in:
Patrick von Platen 2020-03-12 21:10:51 +01:00
parent f1c71da115
commit 6a82f774f2

View File

@ -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,
)