Skip to content

GitLab

Unverified Commit 3bbe68f8 authored by Lysandre Debut's avatar Lysandre Debut Committed by GitHub
Browse files

Hubert test fix (#13261)

parent 3bb44662
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
tests/test_modeling_hubert.py
View file @ 3bbe68f8
...@@ -563,7 +563,7 @@ class HubertModelIntegrationTest(unittest.TestCase): ...@@ -563,7 +563,7 @@ class HubertModelIntegrationTest(unittest.TestCase):
input_speech = self._load_datasamples(2) 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) input_values = inputs.input_values.to(torch_device)
attention_mask = inputs.attention_mask.to(torch_device) attention_mask = inputs.attention_mask.to(torch_device)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment