Fix HubertRobustTest PT/TF equivalence test on GPU (huggingface#16943)

Co-authored-by: ydshieh <[email protected]>
elusenji · Jun 12, 2022 · 3e5c657 · 3e5c657
1 parent afd312d
commit 3e5c657
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/tests/hubert/test_modeling_hubert.py b/tests/hubert/test_modeling_hubert.py
@@ -133,6 +133,7 @@ def get_config(self):
  hidden_act=self.hidden_act,
  initializer_range=self.initializer_range,
  vocab_size=self.vocab_size,
+ do_stable_layer_norm=self.do_stable_layer_norm,
  )
 
  def create_and_check_model(self, config, input_values, attention_mask):