Hubert test fix (#13261)

3bbe68f8 · Lysandre Debut · GitHub · 3bb44662 · 3bbe68f8
Unverified Commit 3bbe68f8 authored Aug 25, 2021 by Lysandre Debut Committed by GitHub Aug 25, 2021
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

tests/test_modeling_hubert.py tests/test_modeling_hubert.py +1 -1

No files found.
--- a/tests/test_modeling_hubert.py
+++ b/tests/test_modeling_hubert.py
@@ -563,7 +563,7 @@ class HubertModelIntegrationTest(unittest.TestCase):

        input_speech = self._load_datasamples(2)

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

        input_values = inputs.input_values.to(torch_device)
        attention_mask = inputs.attention_mask.to(torch_device)