Hubert test fix (#13261)

2021-08-25 18:41:26 +02:00
parent 3bb4466260
commit 3bbe68f837
1 changed files with 1 additions and 1 deletions
--- 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)