From b518aaf193938247f698a7c4522afe42b025225a Mon Sep 17 00:00:00 2001 From: calpt Date: Fri, 17 Sep 2021 21:36:23 +0200 Subject: [PATCH] Fix GPT2Config parameters in GPT2ModelTester (#13630) --- tests/test_modeling_gpt2.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/tests/test_modeling_gpt2.py b/tests/test_modeling_gpt2.py index a770729303..91d2edcdc8 100644 --- a/tests/test_modeling_gpt2.py +++ b/tests/test_modeling_gpt2.py @@ -141,10 +141,10 @@ class GPT2ModelTester: n_embd=self.hidden_size, n_layer=self.num_hidden_layers, n_head=self.num_attention_heads, - intermediate_size=self.intermediate_size, - hidden_act=self.hidden_act, - hidden_dropout_prob=self.hidden_dropout_prob, - attention_probs_dropout_prob=self.attention_probs_dropout_prob, + n_inner=self.intermediate_size, + activation_function=self.hidden_act, + resid_pdrop=self.hidden_dropout_prob, + attn_pdrop=self.attention_probs_dropout_prob, n_positions=self.max_position_embeddings, n_ctx=self.max_position_embeddings, type_vocab_size=self.type_vocab_size,