Skip to content

GitLab

Unverified Commit 49d5bcb0 authored by Yih-Dar's avatar Yih-Dar Committed by GitHub
Browse files

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


Co-authored-by: default avatarydshieh <ydshieh@users.noreply.github.com>
parent 479fdc49
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
tests/hubert/test_modeling_hubert.py
View file @ 49d5bcb0
......@@ -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):
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment