Fix Bug in FlaxWav2Vec2 Slow Test (#15887)

huggingface · Mar 2, 2022 · 8fd4731 · 8fd4731
1 parent d83d22f
commit 8fd4731
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/wav2vec2/test_modeling_flax_wav2vec2.py b/tests/wav2vec2/test_modeling_flax_wav2vec2.py
@@ -384,7 +384,7 @@ def test_inference_ctc_robust_batched(self):
 
         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