mirror of
https://github.com/huggingface/transformers.git
synced 2025-08-01 18:51:14 +06:00
fix typo
This commit is contained in:
parent
f1c71da115
commit
6a82f774f2
@ -82,7 +82,7 @@ class ModelTester:
|
|||||||
dropout=self.hidden_dropout_prob,
|
dropout=self.hidden_dropout_prob,
|
||||||
attention_dropout=self.attention_probs_dropout_prob,
|
attention_dropout=self.attention_probs_dropout_prob,
|
||||||
max_position_embeddings=self.max_position_embeddings,
|
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,
|
bos_token_id=self.bos_token_id,
|
||||||
pad_token_id=self.pad_token_id,
|
pad_token_id=self.pad_token_id,
|
||||||
)
|
)
|
||||||
|
Loading…
Reference in New Issue
Block a user