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

ydshieh · elusenji · commit 3e5c6571e363 · 2022-06-12T11:14:27.000+02:00
Co-authored-by: ydshieh &lt;ydshieh@users.noreply.github.com&gt;
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):

Original file line number	Diff line number	Diff line change
`@@ -133,6 +133,7 @@ def get_config(self):`
`133`	`133`	`hidden_act=self.hidden_act,`
`134`	`134`	`initializer_range=self.initializer_range,`
`135`	`135`	`vocab_size=self.vocab_size,`
	`136`	`+ do_stable_layer_norm=self.do_stable_layer_norm,`
`136`	`137`	`)`
`137`	`138`
`138`	`139`	`def create_and_check_model(self, config, input_values, attention_mask):`