From c1a138613dacff33364a465d6f7d86ce0822de82 Mon Sep 17 00:00:00 2001 From: Yih-Dar <2521628+ydshieh@users.noreply.github.com> Date: Tue, 31 May 2022 15:01:54 +0200 Subject: [PATCH] Fix ViTMAEModelTester (#17470) Co-authored-by: ydshieh --- tests/models/vit_mae/test_modeling_tf_vit_mae.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/tests/models/vit_mae/test_modeling_tf_vit_mae.py b/tests/models/vit_mae/test_modeling_tf_vit_mae.py index dc73b20e7e..cb54e29b80 100644 --- a/tests/models/vit_mae/test_modeling_tf_vit_mae.py +++ b/tests/models/vit_mae/test_modeling_tf_vit_mae.py @@ -107,6 +107,10 @@ class TFViTMAEModelTester: num_hidden_layers=self.num_hidden_layers, num_attention_heads=self.num_attention_heads, intermediate_size=self.intermediate_size, + decoder_hidden_size=self.hidden_size, + decoder_num_hidden_layers=self.num_hidden_layers, + decoder_num_attention_heads=self.num_attention_heads, + decoder_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,