From 49d5bcb0f3750f09c4f5a5a2c0d7e9ed05dfcb72 Mon Sep 17 00:00:00 2001
From: Yih-Dar <2521628+ydshieh@users.noreply.github.com>
Date: Wed, 27 Apr 2022 10:50:03 +0200
Subject: [PATCH] Fix HubertRobustTest PT/TF equivalence test on GPU (#16943)

Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>
---
 tests/hubert/test_modeling_hubert.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/tests/hubert/test_modeling_hubert.py b/tests/hubert/test_modeling_hubert.py
index 3c3ba96c99..0bc854114d 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):