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

Hubert test fix (#13261)

Browse Source
This commit is contained in:
Lysandre Debut
2021-08-25 18:41:26 +02:00
committed by GitHub
parent 3bb4466260
commit 3bbe68f837
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tests/test_modeling_hubert.py
View File

@@ -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)