Fix Bug in FlaxWav2Vec2 Slow Test (#15887)

8fd47310 · Sanchit Gandhi · GitHub · d83d22f5 · 8fd47310
Unverified Commit 8fd47310 authored Mar 02, 2022 by Sanchit Gandhi Committed by GitHub Mar 02, 2022
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

tests/wav2vec2/test_modeling_flax_wav2vec2.py tests/wav2vec2/test_modeling_flax_wav2vec2.py +1 -1

No files found.
--- a/tests/wav2vec2/test_modeling_flax_wav2vec2.py
+++ b/tests/wav2vec2/test_modeling_flax_wav2vec2.py
@@ -384,7 +384,7 @@ class FlaxWav2Vec2ModelIntegrationTest(unittest.TestCase):

        input_speech = self._load_datasamples(4)

-        inputs = processor(input_speech, return_tensors="pt", padding=True, truncation=True)
+        inputs = processor(input_speech, return_tensors="np", padding=True)

        input_values = inputs.input_values
        attention_mask = inputs.attention_mask