use scale=1.0 in floats_tensor called in speech model testers (#17007)

Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>
This commit is contained in:
Yih-Dar
2022-04-29 14:41:33 +02:00
committed by GitHub
parent e6f00a11d7
commit e952e049b4
12 changed files with 17 additions and 17 deletions

View File

@@ -116,7 +116,7 @@ class Data2VecAudioModelTester:
self.adapter_output_seq_length = (self.output_seq_length - 1) // adapter_stride + 1
def prepare_config_and_inputs(self):
input_values = floats_tensor([self.batch_size, self.seq_length], self.vocab_size)
input_values = floats_tensor([self.batch_size, self.seq_length], scale=1.0)
attention_mask = random_attention_mask([self.batch_size, self.seq_length])
config = self.get_config()