Fix Bug in FlaxWav2Vec2 Slow Test (#15887)

2022-03-02 16:02:26 +01:00
parent d83d22f578
commit 8fd4731072
1 changed files with 1 additions and 1 deletions
--- 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