diff --git a/src/transformers/models/wav2vec2/modeling_tf_wav2vec2.py b/src/transformers/models/wav2vec2/modeling_tf_wav2vec2.py index 39e1539e70a787..78cc2e467ba21f 100644 --- a/src/transformers/models/wav2vec2/modeling_tf_wav2vec2.py +++ b/src/transformers/models/wav2vec2/modeling_tf_wav2vec2.py @@ -1194,8 +1194,8 @@ def input_signature(self): @property def dummy_inputs(self): return { - "input_values": tf.random.uniform(shape=(1, 16000), dtype=tf.float32), - "attention_mask": tf.ones(shape=(1, 16000), dtype=tf.float32), + "input_values": tf.random.uniform(shape=(1, 500), dtype=tf.float32), + "attention_mask": tf.ones(shape=(1, 500), dtype=tf.float32), } def __init__(self, config, *inputs, **kwargs):