From 8fd4731072104462aed8e7da119f25b5137c3d8f Mon Sep 17 00:00:00 2001
From: Sanchit Gandhi <93869735+sanchit-gandhi@users.noreply.github.com>
Date: Wed, 2 Mar 2022 16:02:26 +0100
Subject: [PATCH] Fix Bug in FlaxWav2Vec2 Slow Test (#15887)

---
 tests/wav2vec2/test_modeling_flax_wav2vec2.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tests/wav2vec2/test_modeling_flax_wav2vec2.py b/tests/wav2vec2/test_modeling_flax_wav2vec2.py
index 40579d4fd1..6e25ac8281 100644
--- 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