fix typo
This commit is contained in:
1 parent
f1c71da115
commit
6a82f774f2
1 file changed
+1
-1
@@ -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,
|
||||
)
|
||||
|
||||
Reference in new issue
Block a user