SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

Fix Bug in FlaxWav2Vec2 Slow Test (#15887)

Browse Source
This commit is contained in:
Sanchit Gandhi
2022-03-02 16:02:26 +01:00
committed by GitHub
parent d83d22f578
commit 8fd4731072
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tests/wav2vec2/test_modeling_flax_wav2vec2.py
View File

@@ -384,7 +384,7 @@ class FlaxWav2Vec2ModelIntegrationTest(unittest.TestCase):
input_speech = self._load_datasamples(4) 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 input_values = inputs.input_values
attention_mask = inputs.attention_mask attention_mask = inputs.attention_mask