diff --git a/tests/hubert/test_modeling_hubert.py b/tests/hubert/test_modeling_hubert.py index 3c3ba96c9..0bc854114 100644 --- a/tests/hubert/test_modeling_hubert.py +++ b/tests/hubert/test_modeling_hubert.py @@ -133,6 +133,7 @@ class HubertModelTester: 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):